forked from mindspore-Ecosystem/mindspore
fix gpu hung problem
This commit is contained in:
parent
1f2ca74cd1
commit
3bacea7bba
|
@ -248,7 +248,7 @@ class Iterator:
|
||||||
return self.depipeline.GetNumClasses()
|
return self.depipeline.GetNumClasses()
|
||||||
|
|
||||||
def __deepcopy__(self, memo):
|
def __deepcopy__(self, memo):
|
||||||
return Iterator(copy.deepcopy(self.dataset, memo))
|
return self
|
||||||
|
|
||||||
|
|
||||||
class DictIterator(Iterator):
|
class DictIterator(Iterator):
|
||||||
|
|
Loading…
Reference in New Issue