Browse Source

modify ckpt env

tags/v1.1.0
changzherui 5 years ago
parent
commit
728d2cc765
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      mindspore/_check_version.py

+ 1
- 1
mindspore/_check_version.py View File

@@ -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"



Loading…
Cancel
Save