forked from mindspore-Ecosystem/mindspore
fix dataset deepcopy failed while depipline referenced in the iterator
This commit is contained in:
parent
24fff5fe63
commit
852039b726
|
@ -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):
|
||||
"""
|
||||
|
|
Loading…
Reference in New Issue