Skip to content

Commit 8f66615

Browse files
committed
[LowerConv] skip convs with non-initialized weights
1 parent 84ad7ae commit 8f66615

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

src/qonnx/transformation/lower_convs_to_matmul.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -51,6 +51,10 @@ def apply(self, model):
5151
warnings.warn("Found Conv node with bias, skipping")
5252
continue
5353

54+
if model.get_initializer(node.input[1]) is None:
55+
warnings.warn("Found Conv node with non-initialized weight, skipping")
56+
continue
57+
5458
# extract parameters of node
5559
(
5660
cnv_input,

0 commit comments

Comments
 (0)