修改vgg16训练过程

train_vgg16.py

@@ -90,7 +90,7 @@ def get_args_parser(add_help=True):
     parser = argparse.ArgumentParser(description="PyTorch Classification Training", add_help=add_help)
 
     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("--output-dir", default="checkpoints/vgg16", type=str, help="path to save outputs")
 
     parser.add_argument(
         "-b", "--batch-size", default=2, type=int, help="images per gpu, the total batch size is $NGPU x batch_size"