Remove classifier x 10

pytorch · May 7, 2019 · 7c9f4aa · 7c9f4aa
1 parent c23c1f4
commit 7c9f4aa
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/references/segmentation/train.py b/references/segmentation/train.py
@@ -141,7 +141,7 @@ def main(args):
 
     params_to_optimize = [
         {"params": [p for p in model_without_ddp.backbone.parameters() if p.requires_grad]},
-        {"params": [p for p in model_without_ddp.classifier.parameters() if p.requires_grad], "lr": args.lr * 10},
+        {"params": [p for p in model_without_ddp.classifier.parameters() if p.requires_grad], "lr": args.lr * 1},
     ]
     if args.aux_loss:
         params = [p for p in model_without_ddp.aux_classifier.parameters() if p.requires_grad]