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

mAP calculation during training, if is used flag -map

Browse Source
This commit is contained in:
AlexeyAB
2018-12-09 18:18:47 +03:00
parent 742bb7c7ce
commit dc7f8a32ae
8 changed files with 161 additions and 60 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/convolutional_kernels.cu
View File

@ -295,7 +295,7 @@ void forward_convolutional_layer_gpu(convolutional_layer l, network_state state)
//#ifdef CUDNN_HALF
//if (state.use_mixed_precision) {
int iteration_num = (*state.net.seen) / (state.net.batch*state.net.subdivisions);
if(state.index != 0 && state.net.cudnn_half && !l.xnor && (!state.train || iteration_num > state.net.burn_in))
if (state.index != 0 && state.net.cudnn_half && !l.xnor && (!state.train || iteration_num > state.net.burn_in))
{
//printf("\n CUDNN_HALF!!! state.index = %d \n", state.index);