From 0305fb4d99cf1efc7d4aa4d2ee2d65d54500d437 Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Thu, 26 Nov 2015 19:48:01 +0000
Subject: [PATCH] Some changes
---
src/activations.c | 7 +++++++
1 files changed, 7 insertions(+), 0 deletions(-)
diff --git a/src/activations.c b/src/activations.c
index d31b1e4..5a62ef5 100644
--- a/src/activations.c
+++ b/src/activations.c
@@ -12,6 +12,8 @@
return "logistic";
case RELU:
return "relu";
+ case ELU:
+ return "elu";
case RELIE:
return "relie";
case RAMP:
@@ -34,6 +36,7 @@
{
if (strcmp(s, "logistic")==0) return LOGISTIC;
if (strcmp(s, "relu")==0) return RELU;
+ if (strcmp(s, "elu")==0) return ELU;
if (strcmp(s, "relie")==0) return RELIE;
if (strcmp(s, "plse")==0) return PLSE;
if (strcmp(s, "linear")==0) return LINEAR;
@@ -53,6 +56,8 @@
return logistic_activate(x);
case RELU:
return relu_activate(x);
+ case ELU:
+ return elu_activate(x);
case RELIE:
return relie_activate(x);
case RAMP:
@@ -84,6 +89,8 @@
return logistic_gradient(x);
case RELU:
return relu_gradient(x);
+ case ELU:
+ return elu_gradient(x);
case RELIE:
return relie_gradient(x);
case RAMP:
--
Gitblit v1.10.0