diff --git a/mindspore/dataset/engine/iterators.py b/mindspore/dataset/engine/iterators.py index d670de508c4..cb56a435f15 100644 --- a/mindspore/dataset/engine/iterators.py +++ b/mindspore/dataset/engine/iterators.py @@ -231,6 +231,9 @@ class Iterator: def num_classes(self): return self.depipeline.GetNumClasses() + def __deepcopy__(self, memo): + return Iterator(copy.deepcopy(self.dataset, memo)) + class DictIterator(Iterator): """