提交 663ae4d7 编写于 作者: M mindspore-ci-bot 提交者: Gitee

!462 fix dataset deepcopy failed while depipline referenced in the iterator

Merge pull request !462 from caojian05/mindspore_master_bugfix
......@@ -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):
"""
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册