!4964 memory ascend while multi model.eval

Merge pull request !4964 from anzhengqi/I1QWT0-memory-ascend
This commit is contained in:
mindspore-ci-bot 2020-08-22 14:21:55 +08:00 committed by Gitee
commit b23b957228
1 changed files with 2 additions and 0 deletions

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):