Merge branch 'style/pytorch2onnx_kneron' into 'Kn_dev_v0.21.1'

style: pytorch2onnx_kneron.py line too long

See merge request eric_wu/mmsegmentation!5
This commit is contained in:
EricWu 2022-04-11 16:01:16 +08:00
commit a783220efa

View File

@ -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: