(self,
rank,
load_save_dict=None,
model_for_opt: tuple = None,
optimizer=None,
global_step=None,
dcp=False,
lora=False,
remove_old_ckpts=True)
| 273 | return None |
| 274 | |
| 275 | def save_ckpt(self, |
| 276 | rank, |
| 277 | load_save_dict=None, |
| 278 | model_for_opt: tuple = None, |
| 279 | optimizer=None, |
| 280 | global_step=None, |
| 281 | dcp=False, |
| 282 | lora=False, |
| 283 | remove_old_ckpts=True): |
| 284 | if remove_old_ckpts: |
| 285 | os.system(f'rm -rf {self.ckpt_dir}/*') |
| 286 | ckpt_dir = os.path.join(self.ckpt_dir, f'{global_step:08d}') |
| 287 | return self.save_ckpt_to_dir(rank, load_save_dict, model_for_opt, |
| 288 | optimizer, ckpt_dir, dcp, lora) |
| 289 | |
| 290 | def save_motion_dict(self, motion_dict, mean, std, device='cuda:0', vis=True, result_folder=None): |
| 291 | if result_folder is None: |
no test coverage detected