(states, is_best, output_dir,
filename='checkpoint.pth')
| 221 | |
| 222 | |
| 223 | def save_checkpoint(states, is_best, output_dir, |
| 224 | filename='checkpoint.pth'): |
| 225 | torch.save(states, os.path.join(output_dir, filename)) |
| 226 | if is_best: |
| 227 | torch.save(states, os.path.join(output_dir, 'checkpoint_best.pth')) |
| 228 | |
| 229 | |
| 230 |
nothing calls this directly
no outgoing calls
no test coverage detected