From 1b5e6d8855064e5d990bd73f1f3a0aa00cbdfb4c Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Sat, 01 Mar 2014 22:41:30 +0000
Subject: [PATCH] Flipping features now a flag
---
src/softmax_layer.c | 5 ++++-
1 files changed, 4 insertions(+), 1 deletions(-)
diff --git a/src/softmax_layer.c b/src/softmax_layer.c
index 1e01bd2..b6b7ff3 100644
--- a/src/softmax_layer.c
+++ b/src/softmax_layer.c
@@ -36,8 +36,11 @@
}
for(i = 0; i < layer.inputs; ++i){
sum += exp(input[i]-largest);
+ //printf("%f, ", input[i]);
}
- sum = largest+log(sum);
+ //printf("\n");
+ if(sum) sum = largest+log(sum);
+ else sum = largest-100;
for(i = 0; i < layer.inputs; ++i){
layer.output[i] = exp(input[i]-sum);
}
--
Gitblit v1.10.0