!4812 Fix server exit error.

Merge pull request !4812 from ZPaC/master-fix-server-exit-error-and-parameter-alloc-for-large-tensor
pull/4812/MERGE
mindspore-ci-bot 5 years ago committed by Gitee
commit b8a66a176e

@ -16,7 +16,6 @@
from collections.abc import Iterable from collections.abc import Iterable
import os import os
import sys
import math import math
import numpy as np import numpy as np
@ -498,7 +497,7 @@ class Model:
list_callback.step_end(run_context) list_callback.step_end(run_context)
if os.getenv("MS_ROLE") == "MS_PSERVER": if os.getenv("MS_ROLE") == "MS_PSERVER":
sys.exit(0) os._exit(0)
should_stop = should_stop or run_context.get_stop_requested() should_stop = should_stop or run_context.get_stop_requested()
if should_stop: if should_stop:
break break

Loading…
Cancel
Save