From 8e04d0d57f6cee5ff0c567c77314b07fafabb469 Mon Sep 17 00:00:00 2001
From: gdh1995 <gdh1995@qq.com>
Date: Sat, 15 Jan 2022 10:01:48 +0800
Subject: [PATCH] fix(export): not throw for unknown structures

---
 torchslim/quantizing/qat_tools.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/torchslim/quantizing/qat_tools.py b/torchslim/quantizing/qat_tools.py
index f1b56d4..227cab5 100644
--- 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":