scripts/fold.py

import sys
import time

import onnxruntime as ort

source, target = sys.argv[1], sys.argv[2]
options = ort.SessionOptions()
options.graph_optimization_level = ort.GraphOptimizationLevel.ORT_ENABLE_BASIC
options.optimized_model_filepath = target
started = time.time()
ort.InferenceSession(source, options, providers=["CPUExecutionProvider"])
print(f"folded in {time.time() - started:.1f} s -> {target}")