From d0b9326a352ed2fbc3ae66fdef40b4533a2f211d Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Tue, 11 Aug 2015 06:22:27 +0000
Subject: [PATCH] Hacks to get nightmare to not break gridsizing
---
src/activation_kernels.cu | 12 ++++++++++++
1 files changed, 12 insertions(+), 0 deletions(-)
diff --git a/src/activation_kernels.cu b/src/activation_kernels.cu
index 32c032c..16dd4d2 100644
--- a/src/activation_kernels.cu
+++ b/src/activation_kernels.cu
@@ -6,7 +6,9 @@
__device__ float linear_activate_kernel(float x){return x;}
__device__ float logistic_activate_kernel(float x){return 1./(1. + exp(-x));}
__device__ float relu_activate_kernel(float x){return x*(x>0);}
+__device__ float relie_activate_kernel(float x){return x*(x>0);}
__device__ float ramp_activate_kernel(float x){return x*(x>0)+.1*x;}
+__device__ float leaky_activate_kernel(float x){return (x>0) ? x : .1*x;}
__device__ float tanh_activate_kernel(float x){return (exp(2*x)-1)/(exp(2*x)+1);}
__device__ float plse_activate_kernel(float x)
{
@@ -18,7 +20,9 @@
__device__ float linear_gradient_kernel(float x){return 1;}
__device__ float logistic_gradient_kernel(float x){return (1-x)*x;}
__device__ float relu_gradient_kernel(float x){return (x>0);}
+__device__ float relie_gradient_kernel(float x){return (x>0) ? 1 : .01;}
__device__ float ramp_gradient_kernel(float x){return (x>0)+.1;}
+__device__ float leaky_gradient_kernel(float x){return (x>0) ? 1 : .1;}
__device__ float tanh_gradient_kernel(float x){return 1-x*x;}
__device__ float plse_gradient_kernel(float x){return (x < 0 || x > 1) ? .01 : .125;}
@@ -31,8 +35,12 @@
return logistic_activate_kernel(x);
case RELU:
return relu_activate_kernel(x);
+ case RELIE:
+ return relie_activate_kernel(x);
case RAMP:
return ramp_activate_kernel(x);
+ case LEAKY:
+ return leaky_activate_kernel(x);
case TANH:
return tanh_activate_kernel(x);
case PLSE:
@@ -50,8 +58,12 @@
return logistic_gradient_kernel(x);
case RELU:
return relu_gradient_kernel(x);
+ case RELIE:
+ return relie_gradient_kernel(x);
case RAMP:
return ramp_gradient_kernel(x);
+ case LEAKY:
+ return leaky_gradient_kernel(x);
case TANH:
return tanh_gradient_kernel(x);
case PLSE:
--
Gitblit v1.10.0