diff --git a/tools/pytorch2onnx.py b/tools/pytorch2onnx.py
index 578b48fa8c..71f1bb7227 100644
--- 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')