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

fix block.export #17970

Merged
merged 9 commits into from Sep 1, 2020
Merged

fix block.export #17970

Changes from all commits
Commits
Show all changes
9 commits
Select commit Hold shift + click to select a range
1c5a325
fix block.export
chinakook Apr 4, 2020
da18626
Update block.py
chinakook Apr 6, 2020
8e5e618
add allow_extra
chinakook Apr 6, 2020
b75bedf
Update block.py
chinakook May 22, 2020
e702da9
Update python/mxnet/gluon/block.py
chinakook May 23, 2020
acb8418
Update block.py
chinakook Aug 21, 2020
3a4cb53
Merge branch 'master' into patch-6
chinakook Aug 31, 2020
7e8b2e7
Update block.py
chinakook Sep 1, 2020
5df00db
Update python/mxnet/gluon/block.py
chinakook Sep 1, 2020
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Jump to
Jump to file
Failed to load files.
Diff view
Diff view
7 changes: 5 additions & 2 deletions python/mxnet/gluon/block.py
View file
Open in desktop
Expand Up @@ -1348,8 +1348,11 @@ def export(self, path, epoch=0, remove_amp_cast=True):
if name in arg_names:
arg_dict['arg:{}'.format(name)] = param._reduce()
else:
assert name in aux_names
arg_dict['aux:{}'.format(name)] = param._reduce()
if name not in aux_names:
warnings.warn('Parameter "{name}" is not found in the graph. '
.format(name=name), stacklevel=3)
else:
arg_dict['aux:%s'%name] = param._reduce()
save_fn = _mx_npx.save if is_np_array() else ndarray.save
params_filename = '%s-%04d.params'%(path, epoch)
save_fn(params_filename, arg_dict)
Expand Down