(model, cfg)
| 18 | |
| 19 | |
| 20 | def build_optimizer(model, cfg): |
| 21 | optimizer_cfg = copy.deepcopy(cfg) |
| 22 | constructor_type = optimizer_cfg.pop("constructor", "DefaultOptimizerConstructor") |
| 23 | paramwise_cfg = optimizer_cfg.pop("paramwise_cfg", None) |
| 24 | optim_constructor = build_optimizer_constructor( |
| 25 | dict(type=constructor_type, optimizer_cfg=optimizer_cfg, paramwise_cfg=paramwise_cfg) |
| 26 | ) |
| 27 | optimizer = optim_constructor(model) |
| 28 | return optimizer |
no test coverage detected