From 728d2cc765f755851ba2499d4acfb38a3d65dac3 Mon Sep 17 00:00:00 2001 From: changzherui Date: Tue, 27 Oct 2020 12:33:45 +0800 Subject: [PATCH] modify ckpt env --- mindspore/_check_version.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mindspore/_check_version.py b/mindspore/_check_version.py index 8fff54c81b..f710196d38 100644 --- a/mindspore/_check_version.py +++ b/mindspore/_check_version.py @@ -288,7 +288,7 @@ def _set_pb_env(): "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": + elif os.getenv("PROTOCOL_BUFFERS_PYTHON_IMPLEMENTATION") is None: logger.warning("Set the env `PROTOCOL_BUFFERS_PYTHON_IMPLEMENTATION=python` to prevent memory overflow.") os.environ["PROTOCOL_BUFFERS_PYTHON_IMPLEMENTATION"] = "python"