p47068915
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

!12592 fix checkpoint cache and prosess exception 

From: @fangzehua
Reviewed-by: @kisnwang,@zhoufeng54
Signed-off-by: @zhoufeng54
This commit is contained in:
mindspore-ci-bot 2021-02-25 11:01:47 +08:00 committed by Gitee
parent 50d45e746c da85c804bc
commit 52fac12367
2 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
mindspore/ccsrc/backend/session/executor.cc
View File

@ -349,6 +349,7 @@ void Executor::WaitTaskGraphAvailable(const SessionPtr &session, const std::shar
mindspore::ScopedLongRunning long_running;
for (auto &tensor : task->input_tensors_) {
if (tensor->NeedWait() && !tensor->IsGraphOutput()) {
MsException::Instance().CheckException();
tensor->Wait();
}
}

6
mindspore/train/callback/_checkpoint.py
View File

@ -296,9 +296,6 @@ class ModelCheckpoint(Callback):
cb_params = run_context.original_args()
_to_save_last_ckpt = True
# if param is cache enable, flush data from cache to host before epoch end
self._flush_from_cache(cb_params)
self._save_ckpt(cb_params, _to_save_last_ckpt)
thread_list = threading.enumerate()
@ -328,6 +325,9 @@ class ModelCheckpoint(Callback):
if cb_params.cur_step_num == self._last_triggered_step:
return
# if param is cache enable, flush data from cache to host before save_ckpt
self._flush_from_cache(cb_params)
save_ckpt = self._check_save_ckpt(cb_params, force_to_save)
step_num_in_epoch = int((cb_params.cur_step_num - 1) % cb_params.batch_num + 1)