(args, iteration, optimizer)
| 213 | |
| 214 | |
| 215 | def save_zero_checkpoint(args, iteration, optimizer): |
| 216 | zero_sd = {'iteration': iteration, |
| 217 | 'optimizer_state_dict': optimizer.state_dict()} |
| 218 | zero_checkpoint_name = get_checkpoint_name(args.save, iteration, zero=True) |
| 219 | ensure_directory_exists(zero_checkpoint_name) |
| 220 | torch.save(zero_sd, zero_checkpoint_name) |
| 221 | print(' successfully saved {}'.format(zero_checkpoint_name)) |
| 222 | |
| 223 | |
| 224 | def save_checkpoint(iteration, model, optimizer, lr_scheduler, args, tag=None, barrier=True, |
nothing calls this directly
no test coverage detected