Skip to content

Commit c42e4e1

Browse files
committed
fix validate
1 parent 2271777 commit c42e4e1

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

onnx_diagnostic/torch_models/validate.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -830,7 +830,7 @@ def node_iter(proto):
830830
yield node
831831
for att in node.attribute:
832832
if att.type == onnx.AttributeProto.GRAPH:
833-
yield from att.g
833+
yield from node_iter(att.g)
834834
if hasattr(proto, "initializer"):
835835
yield from proto.initializer
836836
else:

0 commit comments

Comments
 (0)