Searched refs:group_ckpt_save_file (Results 1 – 5 of 5) sorted by relevance
120 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
141 void ParallelContext::set_group_ckpt_save_file(const std::string &group_ckpt_save_file) { in set_group_ckpt_save_file() argument142 group_ckpt_save_file_ = group_ckpt_save_file; in set_group_ckpt_save_file()
336 def set_group_ckpt_save_file(self, group_ckpt_save_file): argument340 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,
37 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()
79 group_ckpt_save_file="./group_stage1.ckpt")142 group_ckpt_save_file="./group_stage1.ckpt")