Searched refs:strategy_ckpt_save_file (Results 1 – 9 of 9) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/ |
| D | context.h | 118 void set_strategy_ckpt_save_file(const std::string &strategy_ckpt_save_file);
119 std::string strategy_ckpt_save_file() const { return strategy_ckpt_save_file_; } in strategy_ckpt_save_file() function
|
| D | context.cc | 137 void ParallelContext::set_strategy_ckpt_save_file(const std::string &strategy_ckpt_save_file) { in set_strategy_ckpt_save_file() argument
138 strategy_ckpt_save_file_ = strategy_ckpt_save_file; in set_strategy_ckpt_save_file()
|
| /third_party/mindspore/mindspore/parallel/ |
| D | _auto_parallel_context.py | 317 def set_strategy_ckpt_save_file(self, strategy_ckpt_save_file): argument
326 dir_path = os.path.dirname(strategy_ckpt_save_file)
329 self._context_handle.set_strategy_ckpt_save_file(strategy_ckpt_save_file)
683 strategy_ckpt_save_file=str, full_batch=bool, enable_parallel_optimizer=bool,
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/strategy_checkpoint/ |
| D | parallel_strategy_checkpoint.cc | 35 instance.save_file_ = ParallelContext::GetInstance()->strategy_ckpt_save_file(); in GetInstance()
36 … instance.save_checkpoint_on_ = !ParallelContext::GetInstance()->strategy_ckpt_save_file().empty(); in GetInstance()
|
| /third_party/mindspore/tests/st/model_zoo_tests/wide_and_deep/python_file_for_ci/ |
| D | train_and_test_multinpu_ci.py | 98 context.set_auto_parallel_context(strategy_ckpt_save_file="./strategy_train.ckpt")
|
| D | callbacks.py | 112 context.set_auto_parallel_context(strategy_ckpt_save_file="",
|
| /third_party/mindspore/tests/ut/python/parallel/ |
| D | test_strategy_checkpoint.py | 78 …set_auto_parallel_context(device_num=8, global_rank=0, strategy_ckpt_save_file="./strategy_stage1.…
206 …set_auto_parallel_context(device_num=8, global_rank=0, strategy_ckpt_save_file="./strategy_stage1_…
|
| /third_party/mindspore/mindspore/ |
| D | context.py | 347 strategy_ckpt_save_file=str, full_batch=bool, enable_parallel_optimizer=bool,
|
| /third_party/mindspore/mindspore/ccsrc/pipeline/jit/ |
| D | init.cc | 176 ….def("get_strategy_ckpt_save_file", &ParallelContext::strategy_ckpt_save_file, "Get strategy check… in PYBIND11_MODULE()
|