提交 78d0ced8 编写于 作者: M mindspore-ci-bot 提交者: Gitee

!3304 fix dataset iterator bug

Merge pull request !3304 from jinyaohui/master
......@@ -222,9 +222,9 @@ class _DatasetIterNormal:
self.dataset = dataset
self.device_num = _get_device_num()
self.global_rank = _get_global_rank()
self.iter = self.dataset.create_tuple_iterator()
def __iter__(self):
self.iter = self.dataset.create_tuple_iterator()
return self
def __next__(self):
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册