diff --git a/mindspore/train/model.py b/mindspore/train/model.py index d4dd29bd5d..324b09b453 100755 --- a/mindspore/train/model.py +++ b/mindspore/train/model.py @@ -16,7 +16,6 @@ from collections.abc import Iterable import os -import sys import math import numpy as np @@ -498,7 +497,7 @@ class Model: list_callback.step_end(run_context) if os.getenv("MS_ROLE") == "MS_PSERVER": - sys.exit(0) + os._exit(0) should_stop = should_stop or run_context.get_stop_requested() if should_stop: break