From 9062ea4bdd4de6ee5971a00248968827171c9358 Mon Sep 17 00:00:00 2001 From: ougongchang Date: Fri, 10 Jul 2020 14:27:09 +0800 Subject: [PATCH] There is a error in the SummaryCollector example. The useage is error when collect custom lineage data. --- mindspore/train/callback/_summary_collector.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/mindspore/train/callback/_summary_collector.py b/mindspore/train/callback/_summary_collector.py index 1550c3c55..ded0e9a65 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) """ -- GitLab