(model, optimizer, filename="my_checkpoint.pth.tar")
| 50 | |
| 51 | |
| 52 | def save_checkpoint(model, optimizer, filename="my_checkpoint.pth.tar"): |
| 53 | print("=> Saving checkpoint") |
| 54 | checkpoint = { |
| 55 | "state_dict": model.state_dict(), |
| 56 | "optimizer": optimizer.state_dict(), |
| 57 | } |
| 58 | torch.save(checkpoint, filename) |
| 59 | |
| 60 | |
| 61 | def load_checkpoint(checkpoint_file, model, optimizer, lr): |