From d3ada156737b03ba4722ee3e11e2798e7947e768 Mon Sep 17 00:00:00 2001 From: ougongchang Date: Mon, 22 Jun 2020 17:21:04 +0800 Subject: [PATCH] Change the attribute to children, becuase the attribute has beed changed in dataset --- mindspore/train/callback/_summary_collector.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mindspore/train/callback/_summary_collector.py b/mindspore/train/callback/_summary_collector.py index e2e4a9cc2d..f86c1a2d32 100644 --- a/mindspore/train/callback/_summary_collector.py +++ b/mindspore/train/callback/_summary_collector.py @@ -677,7 +677,7 @@ class SummaryCollector(Callback): return output_dataset.dataset_dir if isinstance(output_dataset, dataset_files_set): return output_dataset.dataset_files[0] - return self._get_dataset_path(output_dataset.input[0]) + return self._get_dataset_path(output_dataset.children[0]) @staticmethod def _get_ckpt_file_path(cb_params):