Skip to content

Commit 31825a7

Browse files
committed
Add Hardswish #60
1 parent 3943a43 commit 31825a7

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

onnx2pytorch/convert/operations.py

+2
Original file line numberDiff line numberDiff line change
@@ -238,6 +238,8 @@ def convert_operations(onnx_graph, opset_version, batch_dim=0, enable_pruning=Tr
238238
op = nn.Sigmoid()
239239
elif node.op_type == "HardSigmoid":
240240
op = Hardsigmoid(**extract_attributes(node))
241+
elif node.op_type == "HardSwish":
242+
op = nn.Hardswish()
241243
elif node.op_type == "Slice":
242244
op = Slice(**extract_attributes(node))
243245
elif node.op_type == "Softmax":

0 commit comments

Comments
 (0)