fix convbn fusion bug

inisis · Nov 12, 2024 · 0e51815 · 0e51815
1 parent 8c82f56
commit 0e51815
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/onnxslim/core/pattern/fusion/convbn.py b/onnxslim/core/pattern/fusion/convbn.py
@@ -29,7 +29,7 @@ def rewrite(self, opset=11):
         conv_transpose_node = self.conv_0
         conv_transpose_node_users = get_node_users(conv_transpose_node)
         node = self.bn_0
-        if len(conv_transpose_node_users) == 1:
+        if len(conv_transpose_node_users) == 1 and all([isinstance(value, gs.Constant) for value in node.inputs[1:]]):
             conv_transpose_weight = conv_transpose_node.inputs[1].values
             bn_node = node
             bn_scale = bn_node.inputs[1].values