From 90d354a2a5a3ba76071337d8794cfc00f7bc5fab Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Sat, 13 Dec 2014 20:01:21 +0000
Subject: [PATCH] fixed dropout ><
---
src/network_gpu.c | 4 ++++
1 files changed, 4 insertions(+), 0 deletions(-)
diff --git a/src/network_gpu.c b/src/network_gpu.c
index 163d914..d09aa71 100644
--- a/src/network_gpu.c
+++ b/src/network_gpu.c
@@ -101,6 +101,10 @@
maxpool_layer layer = *(maxpool_layer *)net.layers[i];
backward_maxpool_layer_gpu(layer, prev_delta);
}
+ else if(net.types[i] == DROPOUT){
+ dropout_layer layer = *(dropout_layer *)net.layers[i];
+ backward_dropout_layer_gpu(layer, prev_delta);
+ }
else if(net.types[i] == SOFTMAX){
softmax_layer layer = *(softmax_layer *)net.layers[i];
backward_softmax_layer_gpu(layer, prev_delta);
--
Gitblit v1.10.0