test.py 437 Bytes
Newer Older
zk's avatar
zk committed
1
2
3
4
5
6
7
8
9
10
11
12
13
import onnx

model = onnx.load("weights/ground.onnx")

for vi in model.graph.value_info:
    dims = [d.dim_value for d in vi.type.tensor_type.shape.dim]
    if any(d == 0 for d in dims):
        print("⚠️ zero dim in value_info:", vi.name, dims)

for vi in model.graph.output:
    dims = [d.dim_value for d in vi.type.tensor_type.shape.dim]
    if any(d == 0 for d in dims):
        print("⚠️ zero dim in output:", vi.name, dims)