Browse Source

modify logging message

tags/v1.1.0
changzherui 5 years ago
parent
commit
59f6beb6a2
2 changed files with 4 additions and 2 deletions
  1. +2
    -1
      mindspore/_check_version.py
  2. +2
    -1
      mindspore/train/serialization.py

+ 2
- 1
mindspore/_check_version.py View File

@@ -289,7 +289,8 @@ def _set_pb_env():
"it may cause memory limit error durning load checkpoint file. " "it may cause memory limit error durning load checkpoint file. "
"This can be solved by set env `PROTOCOL_BUFFERS_PYTHON_IMPLEMENTATION=python`.") "This can be solved by set env `PROTOCOL_BUFFERS_PYTHON_IMPLEMENTATION=python`.")
elif os.getenv("PROTOCOL_BUFFERS_PYTHON_IMPLEMENTATION") is None: elif os.getenv("PROTOCOL_BUFFERS_PYTHON_IMPLEMENTATION") is None:
logger.warning("Set the env `PROTOCOL_BUFFERS_PYTHON_IMPLEMENTATION=python` to prevent memory overflow.")
logger.warning("Setting the env `PROTOCOL_BUFFERS_PYTHON_IMPLEMENTATION=python` to prevent memory overflow "
"during save or load checkpoint file.")
os.environ["PROTOCOL_BUFFERS_PYTHON_IMPLEMENTATION"] = "python" os.environ["PROTOCOL_BUFFERS_PYTHON_IMPLEMENTATION"] = "python"






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

@@ -379,7 +379,8 @@ def load_param_into_net(net, parameter_dict, strict_load=False):
for param_name in param_not_load: for param_name in param_not_load:
logger.debug("%s", param_name) logger.debug("%s", param_name)


logger.info("Load parameter into net finish, {} parameters has not been loaded.".format(len(param_not_load)))
logger.info("Load parameter into net finish.")
logger.warning("{} parameters in the net are not loaded.".format(len(param_not_load)))
return param_not_load return param_not_load






Loading…
Cancel
Save