Browse Source

!2993 Fix that the usage is error in the SummaryCollector example

Merge pull request !2993 from ougongchang/fix_api
tags/v0.6.0-beta
mindspore-ci-bot Gitee 5 years ago
parent
commit
9b4ec82f93
1 changed files with 4 additions and 2 deletions
  1. +4
    -2
      mindspore/train/callback/_summary_collector.py

+ 4
- 2
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,
>>> # others are not collected
>>> specified = {'collect_metric':True, 'custom_lineage_data': {'version': 'resnet50_v1'}}
>>> specified = {'collect_metric': True}
>>> summary_collector = SummaryCollector('./summary_dir',
>>> 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)
"""



Loading…
Cancel
Save