()
| 112 | |
| 113 | |
| 114 | def main(): |
| 115 | args, cfg = get_args() |
| 116 | |
| 117 | if args.onnx or args.trt: |
| 118 | torch2onnx(args, cfg) |
| 119 | print('torch -> onnx: succeess') |
| 120 | |
| 121 | if args.trt: |
| 122 | onnx2trt(args) |
| 123 | print('onnx -> trt: success') |
| 124 | check(args, cfg) |
| 125 | |
| 126 | |
| 127 | if __name__ == '__main__': |
no test coverage detected