mirrors/pytorch-image-models
1
0
Fork 0
mirror of https://github.com/huggingface/pytorch-image-models.git synced 2025-06-03 15:01:08 +08:00
Code Issues Projects Releases Wiki Activity

Merge pull request #302 from Jasha10/create_optimizer-opt_args

Browse Source 
Configure create_optimizer with args.opt_args
This commit is contained in:
Ross Wightman 2020-12-08 10:19:36 -08:00 committed by GitHub
commit 198f6ea0f3
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
timm/optim/optim_factory.py
View File

@ -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]