Browse Source

fix memory ascend while multi model.eval

tags/v0.7.0-beta
anzhengqi 5 years ago
parent
commit
dd942e1807
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