!3304 fix dataset iterator bug

Merge pull request !3304 from jinyaohui/master
2020-07-22 16:39:59 +08:00 · 2020-07-22 16:39:59 +08:00 · 78d0ced83e
parent bc20de741a 8bdba24c5f
commit 78d0ced83e
1 changed files with 1 additions and 1 deletions
--- a/mindspore/train/dataset_helper.py
+++ b/mindspore/train/dataset_helper.py
@ -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):