From 78d5bcac50274af3fe685fdb60ced79521bb7f17 Mon Sep 17 00:00:00 2001
From: zjun <zhangjun0@huawei.com>
Date: Thu, 19 Nov 2020 19:09:50 +0800
Subject: [PATCH] fix onehot bug

Signed-off-by: zjun <zhangjun0@huawei.com>
---
 mindspore/ccsrc/pipeline/pynative/pynative_execute.cc | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/mindspore/ccsrc/pipeline/pynative/pynative_execute.cc b/mindspore/ccsrc/pipeline/pynative/pynative_execute.cc
index 2313cd421b..7c122b77bb 100644
--- a/mindspore/ccsrc/pipeline/pynative/pynative_execute.cc
+++ b/mindspore/ccsrc/pipeline/pynative/pynative_execute.cc
@@ -1972,8 +1972,8 @@ void PynativeExecutor::Clear(const std::string &flag) {
     ConfigManager::GetInstance().ResetIterNum();
     if (top_graph_cells_.find(flag) != top_graph_cells_.end()) {
       op_forward_map_.clear();
-      Clean();
     }
+    Clean();
     return;
   }