|
|
@@ -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"
|
