charlie880624/STDC
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

style: pytorch2onnx_kneron.py line too long

Browse Source
This commit is contained in:
chingning.chen 2022-04-11 15:53:36 +08:00
parent ef31f37c94
commit b324a669c7
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
tools/pytorch2onnx_kneron.py
View File

@ -1,6 +1,5 @@
# All modification made by Kneron Corp.: Copyright (c) 2022 Kneron Corp. # All modification made by Kneron Corp.: Copyright (c) 2022 Kneron Corp.
# Copyright (c) OpenMMLab. All rights reserved. # Copyright (c) OpenMMLab. All rights reserved.
# Original: tools/pytorch2onnx.py, modified by Kneron
import argparse import argparse
import warnings import warnings
@ -279,7 +278,9 @@ def parse_args():
if __name__ == '__main__': if __name__ == '__main__':
args = parse_args() 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) cfg = mmcv.Config.fromfile(args.config)
if args.cfg_options is not None: if args.cfg_options is not None: