Browse Source

!4964 memory ascend while multi model.eval

Merge pull request !4964 from anzhengqi/I1QWT0-memory-ascend
tags/v0.7.0-beta
mindspore-ci-bot Gitee 5 years ago
parent
commit
b23b957228
1 changed files with 2 additions and 0 deletions
  1. +2
    -0
      mindspore/dataset/engine/datasets.py

+ 2
- 0
mindspore/dataset/engine/datasets.py View File

@@ -2535,6 +2535,8 @@ class TransferDataset(DatasetOp):
# need to keep iterator alive so the executionTree is not destroyed
if self._noop_mode():
return
if self.iterator is not None:
del self.iterator
self.iterator = TupleIterator(self, num_epochs=num_epochs)

def stop_send(self):


Loading…
Cancel
Save