diff --git a/mindspore/_check_version.py b/mindspore/_check_version.py index 45c05ed746..e40ff1c937 100644 --- a/mindspore/_check_version.py +++ b/mindspore/_check_version.py @@ -289,7 +289,8 @@ def _set_pb_env(): "it may cause memory limit error durning load checkpoint file. " "This can be solved by set env `PROTOCOL_BUFFERS_PYTHON_IMPLEMENTATION=python`.") 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" diff --git a/mindspore/train/serialization.py b/mindspore/train/serialization.py index 15ba4ab07c..3cbbaa902f 100644 --- a/mindspore/train/serialization.py +++ b/mindspore/train/serialization.py @@ -379,7 +379,8 @@ def load_param_into_net(net, parameter_dict, strict_load=False): for param_name in param_not_load: 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