#!/workspace/kohya_ss/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()) | |