Browse Source

1.Fix server exit error in pytest.

2.Supports large tensor allocation in PS mode.
tags/v0.7.0-beta
ZPaC 5 years ago
parent
commit
e3ee8da68d
1 changed files with 1 additions and 2 deletions
  1. +1
    -2
      mindspore/train/model.py

+ 1
- 2
mindspore/train/model.py View File

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