!2452 Change the dataset attribute in SummaryCollector

Merge pull request !2452 from ougongchang/fix_dataset_bug
5 years ago · ebf30051ff
parent 69d1b4c07d d3ada15673
commit ebf30051ff
1 changed files with 1 additions and 1 deletions
--- 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):