@ -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 5 0 steps TensorSummary .
Default : None , which means the frequency is auto - calculated just to collect at most 2 0 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 = 5 0
default_tensor_summary_limit = 2 0
total_step = cb_params . epoch_num * cb_params . batch_num
self . _collect_tensor_freq = max ( self . _collect_freq , total_step / / default_tensor_summary_limit )