mirror/darknet
1
0
Fork 0
mirror of https://github.com/pjreddie/darknet.git synced 2023-08-10 21:13:14 +03:00
Code Issues Projects Releases Wiki Activity

fixed dropout ><

Browse Source
This commit is contained in:
Joseph Redmon
2014-12-13 12:01:21 -08:00
parent 79fffcce3c
commit 90d354a2a5
8 changed files with 96 additions and 48 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/network.c
View File

@ -219,6 +219,10 @@ void backward_network(network net, float *input)
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);