p41297308
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

There is a error in the SummaryCollector example. 

The useage is error when collect custom lineage data.
This commit is contained in:
ougongchang 2020-07-10 14:27:09 +08:00
parent ee199007ed
commit 9062ea4bdd
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -126,10 +126,12 @@ class SummaryCollector(Callback):
>>> >>>
>>> # Only collect metric, custom lineage data and record data that collected by the summary operator, >>> # Only collect metric, custom lineage data and record data that collected by the summary operator,
>>> # others are not collected >>> # others are not collected
>>> specified = {'collect_metric':True, 'custom_lineage_data': {'version': 'resnet50_v1'}} >>> specified = {'collect_metric': True}
>>> summary_collector = SummaryCollector('./summary_dir', >>> summary_collector = SummaryCollector('./summary_dir',
>>> collect_specified_data=specified, >>> collect_specified_data=specified,
>>> keep_default_action=False) >>> keep_default_action=False,
>>> custom_lineage_data={'version': 'resnet50_v1'}
>>> )
>>> model.train(epoch, dataset, callbacks=summary_collector) >>> model.train(epoch, dataset, callbacks=summary_collector)
""" """