Merge pull request #17540 from YashasSamaga:cuda4dnn-update-mish

5 years ago · c244b456f0
parent 0cbaaba4b1 87ab4ee567
commit c244b456f0
1 changed files with 1 additions and 4 deletions
--- 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);