10 lines
291 B
Python
Executable file
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 caffe2_to_onnx
|
|
|
|
if __name__ == '__main__':
|
|
sys.argv[0] = re.sub(r'(-script\.pyw?|\.exe)?$', '', sys.argv[0])
|
|
sys.exit(caffe2_to_onnx())
|