From 548b931f9d29eb8b9acf9ca5d7dcce7de3776613 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E6=9D=8E=E9=B8=BF=E7=AB=A0?= Date: Sat, 19 Sep 2020 09:31:40 +0800 Subject: [PATCH] flush summary when appropriate --- mindspore/train/callback/_summary_collector.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/mindspore/train/callback/_summary_collector.py b/mindspore/train/callback/_summary_collector.py index 04ab67ffc7..bfd10bb23f 100644 --- a/mindspore/train/callback/_summary_collector.py +++ b/mindspore/train/callback/_summary_collector.py @@ -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: