diff --git a/mindspore/dataset/engine/iterators.py b/mindspore/dataset/engine/iterators.py index d670de508c4c7a16b2e025b4dfd969f692a17de7..cb56a435f153517674290010839ec7c542c47f6b 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): """