diff --git a/references/classification/train.py b/references/classification/train.py index c6bb615e001..fd3e65a5c42 100644 --- a/references/classification/train.py +++ b/references/classification/train.py @@ -441,12 +441,15 @@ def get_args_parser(add_help=True): "--use-deterministic-algorithms", action="store_true", help="Forces the use of deterministic algorithms only." ) parser.add_argument("--interpolation", default="bilinear", help="the interpolation method (default: bilinear)") - parser.add_argument("--val-resize-size", default=256, type=int, - help="the resize size used for validation (default: 256)") - parser.add_argument("--val-crop-size", default=224, type=int, - help="the central crop size used for validation (default: 224)") - parser.add_argument("--train-crop-size", default=224, type=int, - help="the random crop size used for training (default: 224)") + parser.add_argument( + "--val-resize-size", default=256, type=int, help="the resize size used for validation (default: 256)" + ) + parser.add_argument( + "--val-crop-size", default=224, type=int, help="the central crop size used for validation (default: 224)" + ) + parser.add_argument( + "--train-crop-size", default=224, type=int, help="the random crop size used for training (default: 224)" + ) return parser diff --git a/references/classification/train_quantization.py b/references/classification/train_quantization.py index e6124b364f9..b6f52c71632 100644 --- a/references/classification/train_quantization.py +++ b/references/classification/train_quantization.py @@ -237,12 +237,15 @@ def get_args_parser(add_help=True): parser.add_argument("--dist-url", default="env://", help="url used to set up distributed training") parser.add_argument("--interpolation", default="bilinear", help="the interpolation method (default: bilinear)") - parser.add_argument("--val-resize-size", default=256, type=int, - help="the resize size used for validation (default: 256)") - parser.add_argument("--val-crop-size", default=224, type=int, - help="the central crop size used for validation (default: 224)") - parser.add_argument("--train-crop-size", default=224, type=int, - help="the random crop size used for training (default: 224)") + parser.add_argument( + "--val-resize-size", default=256, type=int, help="the resize size used for validation (default: 256)" + ) + parser.add_argument( + "--val-crop-size", default=224, type=int, help="the central crop size used for validation (default: 224)" + ) + parser.add_argument( + "--train-crop-size", default=224, type=int, help="the random crop size used for training (default: 224)" + ) return parser