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

99 problems

Browse Source
This commit is contained in:
Joseph Redmon
2014-12-18 15:46:45 -08:00
parent 19d3ae7267
commit f88baf4a3a
8 changed files with 34 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
src/network.c
View File

@ -74,6 +74,7 @@ void forward_network(network net, float *input, float *truth, int train)
if(!train) continue;
dropout_layer layer = *(dropout_layer *)net.layers[i];
forward_dropout_layer(layer, input);
input = layer.output;
}
else if(net.types[i] == FREEWEIGHT){
if(!train) continue;
@ -119,7 +120,8 @@ float *get_network_output_layer(network net, int i)
softmax_layer layer = *(softmax_layer *)net.layers[i];
return layer.output;
} else if(net.types[i] == DROPOUT){
return get_network_output_layer(net, i-1);
dropout_layer layer = *(dropout_layer *)net.layers[i];
return layer.output;
} else if(net.types[i] == FREEWEIGHT){
return get_network_output_layer(net, i-1);
} else if(net.types[i] == CONNECTED){
@ -153,6 +155,7 @@ float *get_network_delta_layer(network net, int i)
softmax_layer layer = *(softmax_layer *)net.layers[i];
return layer.delta;
} else if(net.types[i] == DROPOUT){
if(i == 0) return 0;
return get_network_delta_layer(net, i-1);
} else if(net.types[i] == FREEWEIGHT){
return get_network_delta_layer(net, i-1);