export_onnx.py

from onnx_modules import export_onnx
import os

if __name__ == "__main__":
    export_path = "BangDreamApi"
    model_path = "Data/V23/models/G_621000.pth"
    config_path = "Data/V23/configs/config.json"
    novq = False
    dev = False
    if not os.path.exists("onnx"):
        os.makedirs("onnx")
    if not os.path.exists(f"onnx/{export_path}"):
        os.makedirs(f"onnx/{export_path}")
    export_onnx(export_path, model_path, config_path, novq, dev)