Browse Source

optimize example

tags/v1.1.0
caozhou 5 years ago
parent
commit
ce35b6c140
1 changed files with 3 additions and 1 deletions
  1. +3
    -1
      mindspore/train/serialization.py

+ 3
- 1
mindspore/train/serialization.py View File

@@ -344,7 +344,9 @@ def load_param_into_net(net, parameter_dict, strict_load=False):
>>> net = Net()
>>> ckpt_file_name = "./checkpoint/LeNet5-1_32.ckpt"
>>> param_dict = load_checkpoint(ckpt_file_name, filter_prefix="conv1")
>>> load_param_into_net(net, param_dict)
>>> param_not_load = load_param_into_net(net, param_dict)
>>> print(param_not_load)
['conv1.weight']
"""
if not isinstance(net, nn.Cell):
logger.error("Failed to combine the net and the parameters.")


Loading…
Cancel
Save