debug1

jasonfan1997 · Sep 30, 2017 · 57dee01 · 57dee01
1 parent 86173a5
commit 57dee01
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/train_placesCNN.py b/train_placesCNN.py
@@ -145,7 +145,7 @@ def main():
     if args.mode >= 2:
         for param in model.module.layer4.parameters():
             param.requires_grad = True
-        optimizer = torch.optim.SGD([model.module.fc.parameters(), model.module.layer4.parameters()], args.lr,
+        optimizer = torch.optim.SGD(list(model.module.fc.parameters())+list(model.module.layer4.parameters()), args.lr,
                                 momentum=args.momentum,
                                 weight_decay=args.weight_decay)
         '''optim.SGD([