Skip to content
This repository has been archived by the owner on Nov 17, 2023. It is now read-only.

Commit

Permalink
Fix pylint
Browse files Browse the repository at this point in the history
  • Loading branch information
anirudh2290 committed Jun 20, 2019
1 parent 65ebc74 commit 7e9633d
Showing 1 changed file with 6 additions and 4 deletions.
10 changes: 6 additions & 4 deletions python/mxnet/gluon/parameter.py
Original file line number Diff line number Diff line change
Expand Up @@ -16,7 +16,7 @@
# under the License.

# coding: utf-8
# pylint: disable=unnecessary-pass
# pylint: disable=unnecessary-pass, too-many-lines
"""Neural network parameter."""
__all__ = ['DeferredInitializationError', 'Parameter', 'Constant',
'ParameterDict', 'tensor_types']
Expand Down Expand Up @@ -957,10 +957,11 @@ def load(self, filename, ctx=None, allow_missing=False,
"with '%s'"%(restore_prefix, name, restore_prefix)
ndarray_load = ndarray.load(filename)
self.load_dict(ndarray_load, ctx, allow_missing,
ignore_extra, restore_prefix, filename, cast_dtype)
ignore_extra, restore_prefix, filename, cast_dtype, dtype_source)

def load_dict(self, param_dict, ctx=None, allow_missing=False,
ignore_extra=False, restore_prefix='', filename=None, cast_dtype=False):
ignore_extra=False, restore_prefix='', filename=None, cast_dtype=False
dtype_source="current"):
"""Load parameters from dict
Parameters
Expand Down Expand Up @@ -1000,4 +1001,5 @@ def load_dict(self, param_dict, ctx=None, allow_missing=False,
"Please make sure source and target networks have the same prefix."%(
name[lprefix:], error_str, _brief_print_list(self._params.keys()))
continue
self[name]._load_init(arg_dict[name], ctx, cast_dtype=cast_dtype, dtype_source=dtype_source)
self[name]._load_init(arg_dict[name], ctx, cast_dtype=cast_dtype,
dtype_source=dtype_source)

0 comments on commit 7e9633d

Please sign in to comment.