improve mish performance and accuracy

pestisid-cmd · Jun 13, 2020 · 87ab4ee · 87ab4ee
1 parent 6fbf704
commit 87ab4ee
Showing 1 changed file with 1 addition and 4 deletions.
diff --git a/modules/dnn/src/cuda/functors.hpp b/modules/dnn/src/cuda/functors.hpp
@@ -54,11 +54,8 @@ struct mish_functor<float> {
         using csl::device::fast_exp;
 
         auto e = fast_exp(value);
-        if (value <= -18.0f)
-            return value * e;
-
         auto n = e * e + 2 * e;
-        if (value <= -5.0f)
+        if (value <= -0.6f)
             return value * fast_divide(n, n + 2);
 
         return value - 2 * fast_divide(value, n + 2);