Explorar o código

修改alexnet训练初始化学习率

liyan hai 7 meses
pai
achega
b0d1325502
Modificáronse 1 ficheiros con 1 adicións e 1 borrados
  1. 1 1
      train_alexnet.py

+ 1 - 1
train_alexnet.py

@@ -137,7 +137,7 @@ def get_args_parser(add_help=True):
     parser.add_argument("--epochs", default=90, type=int, metavar="N", help="number of total epochs to run")
 
     parser.add_argument("--opt", default="sgd", type=str, help="optimizer")
-    parser.add_argument("--lr", default=0.1, type=float, help="initial learning rate")
+    parser.add_argument("--lr", default=0.01, type=float, help="initial learning rate")
     parser.add_argument("--momentum", default=0.9, type=float, metavar="M", help="momentum")
 
     parser.add_argument(