diff --git a/mindspore/ccsrc/frontend/parallel/context.cc b/mindspore/ccsrc/frontend/parallel/context.cc index d4c1bf6edc..732441461b 100644 --- a/mindspore/ccsrc/frontend/parallel/context.cc +++ b/mindspore/ccsrc/frontend/parallel/context.cc @@ -62,6 +62,7 @@ void ParallelContext::Reset() { enable_parallel_optimizer_ = false; all_reduce_fusion_split_indices_.clear(); all_reduce_fusion_split_sizes_.clear(); + strategy_search_mode_ = DYNAMIC_PROGRAMMING; } void ParallelContext::set_device_num(int32_t device_num) { diff --git a/mindspore/parallel/_auto_parallel_context.py b/mindspore/parallel/_auto_parallel_context.py index 9d1dfd09b3..6efa65815b 100644 --- a/mindspore/parallel/_auto_parallel_context.py +++ b/mindspore/parallel/_auto_parallel_context.py @@ -580,5 +580,6 @@ def _reset_auto_parallel_context(): - strategy_ckpt_load_file: "" - strategy_ckpt_save_file: "" - enable_parallel_optimizer: False + - auto_parallel_search_mode: dynamic_programming """ auto_parallel_context().reset()