From cd4776fc329ba6f3cab509def0d60892aed84c3d Mon Sep 17 00:00:00 2001 From: Li Hongzhang Date: Mon, 27 Jul 2020 16:56:07 +0800 Subject: [PATCH] change at-most collected tensor from 50 to 20 When `collect_tensor_freq` is specified as `None`, the `collect_tensor_freq` would be auto calculated. The previous behavior is to collect at most 50 steps, now changing to 20 --- mindspore/train/callback/_summary_collector.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/mindspore/train/callback/_summary_collector.py b/mindspore/train/callback/_summary_collector.py index 4e18c7f2c..17766f140 100644 --- a/mindspore/train/callback/_summary_collector.py +++ b/mindspore/train/callback/_summary_collector.py @@ -110,7 +110,7 @@ class SummaryCollector(Callback): lineage page. In the custom data, the key type support str, and the value type support str/int/float. Default: None, it means there is no custom data. collect_tensor_freq (Optional[int]): Same as the `collect_freq`, but controls TensorSummary specifically. - Default: None, which means the frequency is auto-calculated just to collect at most 50 steps TensorSummary. + Default: None, which means the frequency is auto-calculated just to collect at most 20 steps TensorSummary. max_file_size (Optional[int]): The maximum size in bytes each file can be written to the disk. Default: None, which means no limit. @@ -295,7 +295,7 @@ class SummaryCollector(Callback): self._collect_dataset_graph(cb_params) if self._collect_tensor_freq is None: - default_tensor_summary_limit = 50 + default_tensor_summary_limit = 20 total_step = cb_params.epoch_num * cb_params.batch_num self._collect_tensor_freq = max(self._collect_freq, total_step // default_tensor_summary_limit) -- GitLab