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 chart during training for CUDNN=1

Browse Source
This commit is contained in:
AlexeyAB
2018-12-25 02:34:32 +03:00
parent 78d1ade380
commit f38d060137
3 changed files with 122 additions and 110 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/maxpool_layer.c
View File

@ -104,6 +104,8 @@ void resize_maxpool_layer(maxpool_layer *l, int w, int h)
l->indexes_gpu = cuda_make_int_array(output_size);
l->output_gpu = cuda_make_array(l->output, output_size);
l->delta_gpu = cuda_make_array(l->delta, output_size);
cudnn_maxpool_setup(l);
#endif
}