diff --git a/mindspore/train/dataset_helper.py b/mindspore/train/dataset_helper.py index ea2b8b498f..5cd1a1a36e 100644 --- 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):