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.c | 4 ++++
1 files changed, 4 insertions(+), 0 deletions(-)
diff --git a/src/network.c b/src/network.c
index ae030ce..64a6032 100644
--- a/src/network.c
+++ b/src/network.c
@@ -219,6 +219,10 @@
maxpool_layer layer = *(maxpool_layer *)net.layers[i];
if(i != 0) backward_maxpool_layer(layer, prev_delta);
}
+ else if(net.types[i] == DROPOUT){
+ dropout_layer layer = *(dropout_layer *)net.layers[i];
+ backward_dropout_layer(layer, prev_delta);
+ }
else if(net.types[i] == NORMALIZATION){
normalization_layer layer = *(normalization_layer *)net.layers[i];
if(i != 0) backward_normalization_layer(layer, prev_input, prev_delta);
--
Gitblit v1.10.0