sbjkx
/
mindspore
forked from OSSInnovation/mindspore
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

!6519 Flush the summary writer when appropriate 

Merge pull request !6519 from LiHongzhang/flush_summary
This commit is contained in:
mindspore-ci-bot 2020-09-19 17:45:42 +08:00 committed by Gitee
parent daada0303c 548b931f9d
commit d611fc98dd
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
mindspore/train/callback/_summary_collector.py
View File

@ -321,6 +321,7 @@ class SummaryCollector(Callback):
self._tensor_collect_range = self._get_tensor_collect_range(cb_params, self._dataset_sink_mode)
self._collect_at_step_end(cb_params, plugin_filter=None)
self._first_step = False
self._record.flush()
else:
current = cb_params.cur_epoch_num if self._dataset_sink_mode else cb_params.cur_step_num
if current % self._collect_freq == 0 and current in self._tensor_collect_range:
@ -354,6 +355,9 @@ class SummaryCollector(Callback):
self._collect_histogram(cb_params)
self._record.record(cb_params.cur_step_num, plugin_filter=plugin_filter)
def epoch_end(self, run_context):
self._record.flush()
def end(self, run_context):
cb_params = run_context.original_args()
if cb_params.mode == ModeEnum.TRAIN.value: