From dd942e180700699fccd7c6c00f7d094ad4107c38 Mon Sep 17 00:00:00 2001
From: anzhengqi <anzhengqi.huawei.com>
Date: Sat, 22 Aug 2020 10:34:52 +0800
Subject: [PATCH] fix memory ascend while multi model.eval

---
 mindspore/dataset/engine/datasets.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/mindspore/dataset/engine/datasets.py b/mindspore/dataset/engine/datasets.py
index 26a27448e6..05f828618b 100644
--- a/mindspore/dataset/engine/datasets.py
+++ b/mindspore/dataset/engine/datasets.py
@@ -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):