fix lint

open-mmlab · Apr 12, 2021 · 5ff17c7 · 5ff17c7
1 parent 64757fc
commit 5ff17c7
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/tools/pytorch2onnx.py b/tools/pytorch2onnx.py
@@ -259,7 +259,10 @@ def parse_args():
     parser.add_argument('--checkpoint', help='checkpoint file', default=None)
     parser.add_argument(
         '--input-img', type=str, help='Images for input', default=None)
-    parser.add_argument('--show', action='store_true', help='show onnx graph and segmentation results')
+    parser.add_argument(
+        '--show',
+        action='store_true',
+        help='show onnx graph and segmentation results')
     parser.add_argument(
         '--verify', action='store_true', help='verify the onnx model')
     parser.add_argument('--output-file', type=str, default='tmp.onnx')