diff --git a/timm/optim/optim_factory.py b/timm/optim/optim_factory.py index ecc61c5f..4c0aaca0 100644 --- a/timm/optim/optim_factory.py +++ b/timm/optim/optim_factory.py @@ -57,6 +57,8 @@ def create_optimizer(args, model, filter_bias_and_bn=True): opt_args['eps'] = args.opt_eps if hasattr(args, 'opt_betas') and args.opt_betas is not None: opt_args['betas'] = args.opt_betas + if hasattr(args, 'opt_args') and args.opt_args is not None: + opt_args.update(args.opt_args) opt_split = opt_lower.split('_') opt_lower = opt_split[-1]