Home
last modified time | relevance | path

Searched refs:group_ckpt_save_file (Results 1 – 5 of 5) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/frontend/parallel/
Dcontext.h120 void set_group_ckpt_save_file(const std::string &group_ckpt_save_file);
121 std::string group_ckpt_save_file() const { return group_ckpt_save_file_; } in group_ckpt_save_file() function
Dcontext.cc141 void ParallelContext::set_group_ckpt_save_file(const std::string &group_ckpt_save_file) { in set_group_ckpt_save_file() argument
142 group_ckpt_save_file_ = group_ckpt_save_file; in set_group_ckpt_save_file()
/third_party/mindspore/mindspore/parallel/
D_auto_parallel_context.py336 def set_group_ckpt_save_file(self, group_ckpt_save_file): argument
340 dir_path = os.path.dirname(group_ckpt_save_file)
343 self._context_handle.set_group_ckpt_save_file(group_ckpt_save_file)
684 … grad_accumulation_step=int, all_reduce_fusion_config=list, group_ckpt_save_file=str,
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/strategy_checkpoint/
Dparallel_strategy_checkpoint.cc37 instance.group_info_save_file_ = ParallelContext::GetInstance()->group_ckpt_save_file(); in GetInstance()
38 instance.group_info_save_on_ = !ParallelContext::GetInstance()->group_ckpt_save_file().empty(); in GetInstance()
/third_party/mindspore/tests/ut/python/parallel/
Dtest_strategy_checkpoint.py79 group_ckpt_save_file="./group_stage1.ckpt")
142 group_ckpt_save_file="./group_stage1.ckpt")