From 0f7f2899b65343e56b0a1188f703d459d824d398 Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Mon, 16 Nov 2015 03:51:26 +0000
Subject: [PATCH] Fix for cuda 7.5
---
src/softmax_layer.c | 2 +-
1 files changed, 1 insertions(+), 1 deletions(-)
diff --git a/src/softmax_layer.c b/src/softmax_layer.c
index ea22d05..0d19aca 100644
--- a/src/softmax_layer.c
+++ b/src/softmax_layer.c
@@ -58,7 +58,7 @@
{
int i;
for(i = 0; i < l.inputs*l.batch; ++i){
- state.delta[i] = l.delta[i];
+ state.delta[i] += l.delta[i];
}
}
--
Gitblit v1.10.0