FRIDA-transformed / check_inputs.py
geologist387's picture
Added an onnx model
eef0ae4
raw
history blame contribute delete
248 Bytes
import onnx
model = onnx.load("onnx/frida-onnx/FRIDA.onnx")
for inp in model.graph.input:
shape = [dim.dim_value if dim.dim_value > 0 else str(dim.dim_param) for dim in inp.type.tensor_type.shape.dim]
print(f"Input '{inp.name}': {shape}")