Browse Source

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
tags/v0.7.0-beta
Li Hongzhang 5 years ago
parent
commit
cd4776fc32
1 changed files with 2 additions and 2 deletions
  1. +2
    -2
      mindspore/train/callback/_summary_collector.py

+ 2
- 2
mindspore/train/callback/_summary_collector.py View File

@@ -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)



Loading…
Cancel
Save