From 118bdd6f624a81c7b43689943485f8d70cbd944e Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Fri, 14 Feb 2014 18:26:31 +0000
Subject: [PATCH] Training on VOC
---
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..79375de 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