训练代码去除无关参数

train_alexnet.py

@@ -111,7 +111,6 @@ def get_args_parser(add_help=True):
     parser.add_argument("--data-path", default="dataset/imagenette2-320", type=str, help="dataset path")
     parser.add_argument("--output-dir", default="checkpoints/alexnet", type=str, help="path to save outputs")
 
-    parser.add_argument("--device", default="cuda", type=str, help="device (Use cuda or cpu Default: cuda)")
     parser.add_argument(
         "-b", "--batch-size", default=64, type=int, help="images per gpu, the total batch size is $NGPU x batch_size"
     )

train_vgg16.py

@@ -83,7 +83,6 @@ def get_args_parser(add_help=True):
     parser.add_argument("--data-path", default="dataset/imagenette2-320", type=str, help="dataset path")
     parser.add_argument("--output-dir", default="checkpoints/alexnet", type=str, help="path to save outputs")
 
-    parser.add_argument("--device", default="cuda", type=str, help="device (Use cuda or cpu Default: cuda)")
     parser.add_argument(
         "-b", "--batch-size", default=2, type=int, help="images per gpu, the total batch size is $NGPU x batch_size"
     )