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

fix dataset iterator bug

This commit is contained in:
jinyaohui 2020-07-22 10:04:02 +08:00
parent ca6756b5fe
commit 8bdba24c5f
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/dataset_helper.py
View File

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