OrgLion-ML
/
darknet
mirror of https://github.com/AlexeyAB/darknet.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Compile fix

This commit is contained in:
AlexeyAB 2020-06-15 22:14:01 +03:00
parent d724306ee3
commit e08a818d5f
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/activation_kernels.cu
View File

@ -247,8 +247,8 @@ __device__ float mish_yashas(float x)
__device__ float mish_yashas2(float x) __device__ float mish_yashas2(float x)
{ {
auto e = __expf(x); float e = __expf(x);
auto n = e * e + 2 * e; float n = e * e + 2 * e;
if (x <= -0.6f) if (x <= -0.6f)
return x * __fdividef(n, n + 2); return x * __fdividef(n, n + 2);