(args, iteration, optimizer)
| 177 | |
| 178 | |
| 179 | def save_zero_checkpoint(args, iteration, optimizer): |
| 180 | zero_sd = {'iteration': iteration, |
| 181 | 'optimizer_state_dict': optimizer.state_dict()} |
| 182 | zero_checkpoint_name = get_checkpoint_name(args.save, iteration, zero=True) |
| 183 | ensure_directory_exists(zero_checkpoint_name) |
| 184 | torch.save(zero_sd, zero_checkpoint_name) |
| 185 | print(' successfully saved {}'.format(zero_checkpoint_name)) |
| 186 | |
| 187 | |
| 188 | def save_checkpoint(iteration, model, optimizer, |
nothing calls this directly
no test coverage detected