!4964 memory ascend while multi model.eval

Merge pull request !4964 from anzhengqi/I1QWT0-memory-ascend
pull/4964/MERGE
mindspore-ci-bot 5 years ago committed by Gitee
commit b23b957228

@ -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