(opt, model)
| 76 | |
| 77 | |
| 78 | def set_optimizer(opt, model): |
| 79 | optimizer = optim.SGD(model.parameters(), |
| 80 | lr=opt.learning_rate, |
| 81 | momentum=opt.momentum, |
| 82 | weight_decay=opt.weight_decay) |
| 83 | return optimizer |
| 84 | |
| 85 | |
| 86 | def save_model(model, optimizer, opt, epoch, save_file): |