(args, iteration, optimizer)
| 169 | |
| 170 | |
| 171 | def save_zero_checkpoint(args, iteration, optimizer): |
| 172 | zero_sd = {'iteration': iteration, |
| 173 | 'optimizer_state_dict': optimizer.state_dict()} |
| 174 | zero_checkpoint_name = get_checkpoint_name(args.save, iteration, zero=True) |
| 175 | ensure_directory_exists(zero_checkpoint_name) |
| 176 | torch.save(zero_sd, zero_checkpoint_name) |
| 177 | print(' successfully saved {}'.format(zero_checkpoint_name)) |
| 178 | |
| 179 | def save_checkpoint(iteration, model, optimizer, |
| 180 | lr_scheduler, args): |
nothing calls this directly
no test coverage detected