Browse Source

[bugfix] server process does not exit after training finish

tags/v1.1.0
lizhenyu 5 years ago
parent
commit
9227c212f1
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