From e3ee8da68db82151c9716f9b49a7894bffa2596b Mon Sep 17 00:00:00 2001 From: ZPaC Date: Thu, 20 Aug 2020 14:31:01 +0800 Subject: [PATCH] 1.Fix server exit error in pytest. 2.Supports large tensor allocation in PS mode. --- mindspore/train/model.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) 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