!4964 memory ascend while multi model.eval
Merge pull request !4964 from anzhengqi/I1QWT0-memory-ascend
This commit is contained in:
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…
Reference in New Issue