DA-2-WebGPU / check_onnx_inputs.py
phiph's picture
Upload folder using huggingface_hub
7382c66 verified
import onnx
model = onnx.load("onnx/model.onnx")
print("Input names:")
for input in model.graph.input:
print(f"- {input.name}")
print("\nOutput names:")
for output in model.graph.output:
print(f"- {output.name}")