Browse Source

!10130 [bugfix] server process does not exit after training finish

From: @zyli2020
Reviewed-by: @limingqi107,@cristoval
Signed-off-by: @limingqi107,@cristoval
tags/v1.1.0
mindspore-ci-bot Gitee 5 years ago
parent
commit
64bf4a211f
1 changed files with 2 additions and 0 deletions
  1. +2
    -0
      mindspore/train/model.py

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

@@ -456,6 +456,8 @@ class Model:
cb_params.cur_step_num += 1
cb_params.net_outputs = outputs
list_callback.step_end(run_context)
if _is_role_pserver():
os._exit(0)

dataset_helper.continue_send()
list_callback.epoch_end(run_context)


Loading…
Cancel
Save