From 1aec025cba5ea68d7c97b358c6c658a2ea7237b0 Mon Sep 17 00:00:00 2001 From: changzherui Date: Mon, 26 Oct 2020 15:34:57 +0800 Subject: [PATCH] fix ckpt env bug --- mindspore/_check_version.py | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/mindspore/_check_version.py b/mindspore/_check_version.py index 5d42a5c78f..8fff54c81b 100644 --- a/mindspore/_check_version.py +++ b/mindspore/_check_version.py @@ -284,10 +284,11 @@ def check_version_and_env_config(): def _set_pb_env(): """Set env variable `PROTOCOL_BUFFERS` to prevent memory overflow.""" if os.getenv("PROTOCOL_BUFFERS_PYTHON_IMPLEMENTATION") == "cpp": - logger.warning("Current env variable `PROTOCOL_BUFFERS_PYTHON_IMPLEMENTATION=cpp`,\ - When the parameter is too large, it may cause memory limit error.\ - This can be solved by set env `PROTOCOL_BUFFERS_PYTHON_IMPLEMENTATION=python`.") - elif os.getenv("PROTOCOL_BUFFERS_PYTHON_IMPLEMENTATION") == "": + logger.warning("Current env variable `PROTOCOL_BUFFERS_PYTHON_IMPLEMENTATION=cpp`. " + "When the checkpoint file is too large, " + "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") != "python": logger.warning("Set the env `PROTOCOL_BUFFERS_PYTHON_IMPLEMENTATION=python` to prevent memory overflow.") os.environ["PROTOCOL_BUFFERS_PYTHON_IMPLEMENTATION"] = "python"