(self,
rank,
load_save_dict: dict = None,
model_for_opt: tuple = None,
optimizer=None,
global_step: int = 0,
load_optimizer_only=False,
lora=False)
| 236 | return None |
| 237 | |
| 238 | def load_ckpt(self, |
| 239 | rank, |
| 240 | load_save_dict: dict = None, |
| 241 | model_for_opt: tuple = None, |
| 242 | optimizer=None, |
| 243 | global_step: int = 0, |
| 244 | load_optimizer_only=False, |
| 245 | lora=False): |
| 246 | ckpt_dir = os.path.join(self.ckpt_dir, f'{global_step:08d}') |
| 247 | return self.load_ckpt_from_dir(rank, load_save_dict, model_for_opt, |
| 248 | optimizer, ckpt_dir, |
| 249 | load_optimizer_only, |
| 250 | lora) |
| 251 | |
| 252 | @staticmethod |
| 253 | def save_ckpt_to_dir(rank, |
no test coverage detected