Fixing linter.

pytorch · Oct 21, 2021 · bf9bfb6 · bf9bfb6
1 parent 9ee69c4
commit bf9bfb6
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 12 deletions.
diff --git 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
@@ -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