|
@@ -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("--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("--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("--momentum", default=0.9, type=float, metavar="M", help="momentum")
|
|
|
|
|
|
parser.add_argument(
|
|
parser.add_argument(
|