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

Commit

Permalink
[DOC] Fix tutorial link, and better error msg (#17057)
Browse files Browse the repository at this point in the history
  • Loading branch information
eric-haibin-lin authored Dec 14, 2019
1 parent f045018 commit 042682e
Showing 1 changed file with 10 additions and 4 deletions.
14 changes: 10 additions & 4 deletions python/mxnet/gluon/parameter.py
Original file line number Diff line number Diff line change
Expand Up @@ -972,7 +972,7 @@ def save(self, filename, strip_prefix=''):
"this may be due to your Block shares parameters from other "
"Blocks or you forgot to use 'with name_scope()' when creating "
"child blocks. For more info on naming, please see "
"https://mxnet.apache.org/api/python/docs/tutorials/packages/gluon/naming.html"%(
"https://mxnet.io/api/python/docs/tutorials/packages/gluon/blocks/naming.html"%(
strip_prefix, param.name, strip_prefix))
arg_dict[param.name[len(strip_prefix):]] = weight
ndarray.save(filename, arg_dict)
Expand Down Expand Up @@ -1006,7 +1006,9 @@ def load(self, filename, ctx=None, allow_missing=False,
for name in self.keys():
assert name.startswith(restore_prefix), \
"restore_prefix is '%s' but Parameters name '%s' does not start " \
"with '%s'"%(restore_prefix, name, restore_prefix)
"with '%s'. For more info on naming, please see " \
"https://mxnet.io/api/python/docs/tutorials/packages/gluon/blocks/naming.html"%(
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, dtype_source)
Expand Down Expand Up @@ -1043,14 +1045,18 @@ def load_dict(self, param_dict, ctx=None, allow_missing=False,
for name in self.keys():
assert name in arg_dict, \
"Parameter '%s' is missing in %s, which contains parameters: %s. " \
"Please make sure source and target networks have the same prefix."%(
"Please make sure source and target networks have the same prefix." \
"For more info on naming, please see " \
"https://mxnet.io/api/python/docs/tutorials/packages/gluon/blocks/naming.html"%(
name[lprefix:], error_str, _brief_print_list(arg_dict.keys()))
for name in arg_dict:
if name not in self._params:
assert ignore_extra, \
"Parameter '%s' loaded from %s is not present in ParameterDict, " \
"choices are: %s. Set ignore_extra to True to ignore. " \
"Please make sure source and target networks have the same prefix."%(
"Please make sure source and target networks have the same prefix." \
"For more info on naming, please see " \
"https://mxnet.io/api/python/docs/tutorials/packages/gluon/blocks/naming.html"%(
name[lprefix:], error_str, _brief_print_list(self._params.keys()))
continue
self[name]._load_init(arg_dict[name], ctx, cast_dtype=cast_dtype,
Expand Down

0 comments on commit 042682e

Please sign in to comment.