cfg/argparse to kwargs helper Convert optimizer args in argparse args or cfg like object to keyword args for updated create fn.
(cfg)
| 42 | |
| 43 | |
| 44 | def optimizer_kwargs(cfg): |
| 45 | """ cfg/argparse to kwargs helper |
| 46 | Convert optimizer args in argparse args or cfg like object to keyword args for updated create fn. |
| 47 | """ |
| 48 | kwargs = dict( |
| 49 | optimizer_name=cfg.opt, |
| 50 | learning_rate=cfg.lr, |
| 51 | weight_decay=cfg.weight_decay, |
| 52 | momentum=cfg.momentum) |
| 53 | if getattr(cfg, 'opt_eps', None) is not None: |
| 54 | kwargs['eps'] = cfg.opt_eps |
| 55 | if getattr(cfg, 'opt_betas', None) is not None: |
| 56 | kwargs['betas'] = cfg.opt_betas |
| 57 | if getattr(cfg, 'opt_args', None) is not None: |
| 58 | kwargs.update(cfg.opt_args) |
| 59 | return kwargs |
| 60 | |
| 61 | |
| 62 | def create_optimizer(args, model, filter_bias_and_bn=True): |
no test coverage detected