Ver Fonte

修改alexnet训练初始化学习率

liyan há 7 meses atrás
pai
commit
b0d1325502
1 ficheiros alterados com 1 adições e 1 exclusões
  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(