Browse Source

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

Merge pull request !462 from caojian05/mindspore_master_bugfix
tags/v0.2.0-alpha
mindspore-ci-bot Gitee 5 years ago
parent
commit
663ae4d7c2
1 changed files with 3 additions and 0 deletions
  1. +3
    -0
      mindspore/dataset/engine/iterators.py

+ 3
- 0
mindspore/dataset/engine/iterators.py View File

@@ -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…
Cancel
Save