Skip to content

M
mindspore

magicwindyyd / mindspore
与 Fork 源项目一致

Fork自 MindSpore / mindspore

通知 1
Star 1
Fork 0
M
mindspore
提交 ebf30051 编写于 作者: M mindspore-ci-bot 提交者： Gitee
浏览文件
操作

!2452 Change the dataset attribute in SummaryCollector 

Merge pull request !2452 from ougongchang/fix_dataset_bug
上级 69d1b4c0 d3ada156
隐藏空白更改
内联 并排
Showing with 1 addition and 1 deletion
mindspore/train/callback/_summary_collector.py
浏览文件 @ ebf30051
...@@ -677,7 +677,7 @@ class SummaryCollector(Callback): ...@@ -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):
......
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或