ONNX.md

Convert Pytorch Model into ONNX Format

usage: convert_to_onnx.py [-h] [--onnx_path ONNX_PATH] [--ckpt_path CKPT_PATH]

Convert Pytorch Model into ONNX Format

optional arguments:
-h, --help            show this help message and exit
--onnx_path ONNX_PATH
                        ONNX file save path
--ckpt_path CKPT_PATH
                        Path to checkpoint file