xxq250
/
mindspore
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

add load ckpt warning

This commit is contained in:
changzherui 2021-05-26 18:05:35 +08:00
parent d5f2fb40f1
commit 772e0d4dec
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
mindspore/train/serialization.py
View File

@ -506,6 +506,8 @@ def load_param_into_net(net, parameter_dict, strict_load=False):
logger.info("Loading parameters into net is finished.")
if param_not_load:
logger.warning("{} parameters in the net are not loaded.".format(len(param_not_load)))
for param_name in param_not_load:
logger.warning("{} is not loaded.".format(param_name))
return param_not_load
@ -641,8 +643,8 @@ def export(net, *inputs, file_name, file_format='AIR', **kwargs):
"""
Export the MindSpore prediction model to a file in the specified format.
Notes:
When exporting to AIR format, the size of a single tensor can not exceed 2GB.
Note:
When exporting to AIR、ONNX format, the size of a single tensor can not exceed 2GB.
Args:
net (Cell): MindSpore network.