fix(export): not throw for unknown structures

THU-MIG · Jan 15, 2022 · 8e04d0d · 8e04d0d
1 parent 235a886
commit 8e04d0d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torchslim/quantizing/qat_tools.py b/torchslim/quantizing/qat_tools.py
@@ -220,7 +220,7 @@ def onnx_post_process(onnx_model):
         if node.op_type == "Conv" or node.op_type == "Add":
             children = get_children(onnx_model, node)
             if len(children) != 1:
-                raise RuntimeError("the number of children node must be 1")
+                continue
             if children[0].op_type == "Relu":
                 continue
             if children[0].op_type == "QuantizeLinear":