From ff7e5842d80ae3399d9533407a36a9e911a3b045 Mon Sep 17 00:00:00 2001 From: Sangdoo Yun Date: Sat, 25 Jul 2020 12:36:28 +0900 Subject: [PATCH] Update optim_factory.py --- timm/optim/optim_factory.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/timm/optim/optim_factory.py b/timm/optim/optim_factory.py index 89d01b03..c6565daa 100644 --- a/timm/optim/optim_factory.py +++ b/timm/optim/optim_factory.py @@ -64,7 +64,7 @@ def create_optimizer(args, model, filter_bias_and_bn=True): optimizer = AdamP( parameters, lr=args.lr, weight_decay=weight_decay, eps=args.opt_eps, delta=0.1, wd_ratio=0.01, nesterov=True) - elif opt_lower == 'sdgp': + elif opt_lower == 'sgdp': optimizer = SGDP( parameters, lr=args.lr, momentum=args.momentum, weight_decay=weight_decay, eps=args.opt_eps, nesterov=True)