Browse Source

!6189 fix deeplabv3 error in pynative

Merge pull request !6189 from chujinjin/fix_deeplabv3_error_in_pynative
tags/v1.0.0
mindspore-ci-bot Gitee 5 years ago
parent
commit
54e22992d7
1 changed files with 2 additions and 1 deletions
  1. +2
    -1
      mindspore/ccsrc/pipeline/pynative/pynative_execute.cc

+ 2
- 1
mindspore/ccsrc/pipeline/pynative/pynative_execute.cc View File

@@ -1468,7 +1468,7 @@ void PynativeExecutor::Clear(const std::string &flag) {
MapClear<std::unordered_map<std::string, FuncGraphPtr>>(&cell_graph_map_, flag);
MapClear<std::unordered_map<std::string, ResourcePtr>>(&cell_resource_map_, flag);
MapClear<std::unordered_map<std::string, FuncGraphPtr>>(&df_builder_map_, flag);
Clean();
// Maybe exit in the pynative runing op, so need reset pynative flag.
auto ms_context = MsContext::GetInstance();
if (ms_context != nullptr) {
@@ -1477,6 +1477,7 @@ 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();
}
return;
}


Loading…
Cancel
Save