pytorch-ai/venv/bin/convert-onnx-to-caffe2
Clemens-Dautermann d2971cf070 added editor
2020-04-25 16:59:20 +02:00

10 lines
291 B
Python
Executable file

#!/home/clemens/repositorys/pytorch-ai/venv/bin/python
# -*- coding: utf-8 -*-
import re
import sys
from caffe2.python.onnx.bin.conversion import onnx_to_caffe2
if __name__ == '__main__':
sys.argv[0] = re.sub(r'(-script\.pyw?|\.exe)?$', '', sys.argv[0])
sys.exit(onnx_to_caffe2())