(file_name, conf, model, optimizer)
| 52 | |
| 53 | |
| 54 | def load_checkpoint(file_name, conf, model, optimizer): |
| 55 | checkpoint = torch.load(file_name) |
| 56 | conf.train.start_epoch = checkpoint["epoch"] |
| 57 | model.load_state_dict(checkpoint["state_dict"]) |
| 58 | optimizer.load_state_dict(checkpoint["optimizer"]) |
| 59 | |
| 60 | |
| 61 | def eval(conf): |