p04896132
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Change the attribute to children, becuase the attribute has beed changed in dataset

This commit is contained in:
ougongchang 2020-06-22 17:21:04 +08:00
parent e726680e38
commit 3c08137904
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mindspore/train/callback/_summary_collector.py
View File

@ -677,7 +677,7 @@ class SummaryCollector(Callback):
return output_dataset.dataset_dir return output_dataset.dataset_dir
if isinstance(output_dataset, dataset_files_set): if isinstance(output_dataset, dataset_files_set):
return output_dataset.dataset_files[0] 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 @staticmethod
def _get_ckpt_file_path(cb_params): def _get_ckpt_file_path(cb_params):