Add Hardswish #60

Talmaj · Sep 15, 2024 · 85d9ea7 · 85d9ea7
1 parent 3943a43
commit 85d9ea7
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/onnx2pytorch/convert/operations.py b/onnx2pytorch/convert/operations.py
@@ -238,6 +238,8 @@ def convert_operations(onnx_graph, opset_version, batch_dim=0, enable_pruning=Tr
             op = nn.Sigmoid()
         elif node.op_type == "HardSigmoid":
             op = Hardsigmoid(**extract_attributes(node))
+        elif node.op_type == "HardSwish":
+            op = nn.Hardswish()
         elif node.op_type == "Slice":
             op = Slice(**extract_attributes(node))
         elif node.op_type == "Softmax":