Typo

yewsiang · Jul 28, 2021 · 4601cee · 4601cee
1 parent f10a4ff
commit 4601cee
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/CUB/train.py b/CUB/train.py
@@ -163,7 +163,7 @@ def train(model, args):
         attr_criterion = None
 
     if args.optimizer == 'Adam':
-        optimizer = torch.optim.Adam(filter(lambda p: p.requires_grad, model.parameters()), lr=arg.lr, weight_decay=args.weight_decay)
+        optimizer = torch.optim.Adam(filter(lambda p: p.requires_grad, model.parameters()), lr=args.lr, weight_decay=args.weight_decay)
     elif args.optimizer == 'RMSprop':
         optimizer = torch.optim.RMSprop(filter(lambda p: p.requires_grad, model.parameters()), lr=args.lr, momentum=0.9, weight_decay=args.weight_decay)
     else: