Untitled

import onnx
from onnxsim import simplify

# Load the exported ONNX model
model = onnx.load("deep_conv_net.onnx")

# Simplify the ONNX model
model_simplified, check = simplify(model)

# Save the simplified model
if check:
    onnx.save(model_simplified, "deep_conv_net_simplified.onnx")
    print("Simplified ONNX model saved successfully.")
else:
    print("Simplification failed.")