diff --git a/mindspore/train/callback/_summary_collector.py b/mindspore/train/callback/_summary_collector.py
index 1550c3c55c..ded0e9a650 100644
--- 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)
     """