From 46bfb4bca9f534c4b84155668bbc162d3a20433c Mon Sep 17 00:00:00 2001
From: nihuini <nihuini@tencent.com>
Date: Mon, 19 Aug 2019 19:13:07 +0800
Subject: [PATCH] write hardswish parameter

---
 tools/onnx/onnx2ncnn.cpp | 13 +++++++++++++
 1 file changed, 13 insertions(+)

diff --git a/tools/onnx/onnx2ncnn.cpp b/tools/onnx/onnx2ncnn.cpp
index c16fc4d2c..c7b5416b8 100644
--- a/tools/onnx/onnx2ncnn.cpp
+++ b/tools/onnx/onnx2ncnn.cpp
@@ -460,6 +460,14 @@ static void fuse_hardswish(onnx::GraphProto* mutable_graph, std::map<std::string
             node4->clear_input();
             node4->add_input(node->input(0));
 
+            onnx::AttributeProto* attr_alpha = node4->add_attribute();
+            attr_alpha->set_name("alpha");
+            attr_alpha->set_f(1.f/6.f);
+
+            onnx::AttributeProto* attr_beta = node4->add_attribute();
+            attr_beta->set_name("beta");
+            attr_beta->set_f(3.f/6.f);
+
             reduced_node_count += 3;
             i += 3;
         }
@@ -1593,6 +1601,11 @@ int main(int argc, char** argv)
         }
         else if (op == "HardSwish")
         {
+            float alpha = get_node_attr_f(node, "alpha", 0.2f);
+            float beta = get_node_attr_f(node, "beta", 0.5f);
+
+            fprintf(pp, " 0=%f", alpha);
+            fprintf(pp, " 1=%f", beta);
         }
         else if (op == "ImageScaler")
         {