(alg, args, inner=False, alias=True, isteacher=False)
| 47 | |
| 48 | |
| 49 | def get_optimizer(alg, args, inner=False, alias=True, isteacher=False): |
| 50 | params = get_params(alg, args, inner, alias, isteacher) |
| 51 | optimizer = torch.optim.SGD( |
| 52 | params, lr=args.lr, momentum=args.momentum, weight_decay=args.weight_decay, nesterov=True) |
| 53 | return optimizer |
| 54 | |
| 55 | |
| 56 | def get_scheduler(optimizer, args): |
no test coverage detected