(file_name, conf, model, optimizer)
| 186 | |
| 187 | |
| 188 | def load_checkpoint(file_name, conf, model, optimizer): |
| 189 | checkpoint = torch.load(file_name) |
| 190 | conf.train.start_epoch = checkpoint["epoch"] |
| 191 | best_performance = checkpoint["best_performance"] |
| 192 | model.load_state_dict(checkpoint["state_dict"]) |
| 193 | optimizer.load_state_dict(checkpoint["optimizer"]) |
| 194 | return best_performance |
| 195 | |
| 196 | |
| 197 | def save_checkpoint(state, file_prefix): |