fix gpu hung problem

This commit is contained in:
xiefangqi 2020-04-27 14:24:23 +08:00
parent 1f2ca74cd1
commit 3bacea7bba
1 changed files with 1 additions and 1 deletions

View File

@ -248,7 +248,7 @@ class Iterator:
return self.depipeline.GetNumClasses()
def __deepcopy__(self, memo):
return Iterator(copy.deepcopy(self.dataset, memo))
return self
class DictIterator(Iterator):