From 8bdba24c5fc858033d3d714707190e0338fd43e6 Mon Sep 17 00:00:00 2001 From: jinyaohui Date: Wed, 22 Jul 2020 10:04:02 +0800 Subject: [PATCH] fix dataset iterator bug --- mindspore/train/dataset_helper.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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):