There is a error in the SummaryCollector example.

The useage is error when collect custom lineage data.
5 years ago · 9062ea4bdd
parent ee199007ed
commit 9062ea4bdd
1 changed files with 4 additions and 2 deletions
--- a/mindspore/train/callback/_summary_collector.py
+++ b/mindspore/train/callback/_summary_collector.py
@ -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)
    """