forked from mindspore-Ecosystem/mindspore
fix onehot bug
Signed-off-by: zjun <zhangjun0@huawei.com>
This commit is contained in:
parent
85e4efd706
commit
78d5bcac50
|
@ -1972,8 +1972,8 @@ void PynativeExecutor::Clear(const std::string &flag) {
|
||||||
ConfigManager::GetInstance().ResetIterNum();
|
ConfigManager::GetInstance().ResetIterNum();
|
||||||
if (top_graph_cells_.find(flag) != top_graph_cells_.end()) {
|
if (top_graph_cells_.find(flag) != top_graph_cells_.end()) {
|
||||||
op_forward_map_.clear();
|
op_forward_map_.clear();
|
||||||
Clean();
|
|
||||||
}
|
}
|
||||||
|
Clean();
|
||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue