From b324a669c7d1060031487165e816bf74d414d8a6 Mon Sep 17 00:00:00 2001 From: "chingning.chen" Date: Mon, 11 Apr 2022 15:53:36 +0800 Subject: [PATCH] style: pytorch2onnx_kneron.py line too long --- tools/pytorch2onnx_kneron.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/tools/pytorch2onnx_kneron.py b/tools/pytorch2onnx_kneron.py index e15cd19..f93b873 100644 --- a/tools/pytorch2onnx_kneron.py +++ b/tools/pytorch2onnx_kneron.py @@ -1,6 +1,5 @@ # All modification made by Kneron Corp.: Copyright (c) 2022 Kneron Corp. # Copyright (c) OpenMMLab. All rights reserved. -# Original: tools/pytorch2onnx.py, modified by Kneron import argparse import warnings @@ -279,7 +278,9 @@ def parse_args(): if __name__ == '__main__': args = parse_args() - assert args.opset_version == 11, "kneron_toolchain currently only supports opset 11" + assert args.opset_version == 11, ( + "kneron_toolchain currently only supports opset 11" + ) cfg = mmcv.Config.fromfile(args.config) if args.cfg_options is not None: