Преглед на файлове

修改alexnet训练初始化学习率

liyan преди 7 месеца
родител
ревизия
b0d1325502
променени са 1 файла, в които са добавени 1 реда и са изтрити 1 реда
  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(