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