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 mAP calculation for CRNN layer during training

Browse Source
This commit is contained in:
AlexeyAB
2019-03-05 02:15:44 +03:00
parent ef803e05df
commit b751bac175
6 changed files with 70 additions and 52 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/crnn_layer.c
View File

@ -263,7 +263,7 @@ void forward_crnn_layer_gpu(layer l, network_state state)
s.train = state.train;
s.workspace = state.workspace;
s.net = state.net;
if(!state.train) s.index = state.index; // use TC only for detection
//if(!state.train) s.index = state.index; // don't use TC at all (especially without cuda_convert_f32_to_f16() )
int i;
layer input_layer = *(l.input_layer);
layer self_layer = *(l.self_layer);