Browse Source

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

tags/v0.6.0-beta
ougongchang 5 years ago
parent
commit
d3ada15673
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      mindspore/train/callback/_summary_collector.py

+ 1
- 1
mindspore/train/callback/_summary_collector.py View File

@@ -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):


Loading…
Cancel
Save