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 random values

Browse Source
This commit is contained in:
AlexeyAB
2019-02-28 16:55:30 +03:00
parent f37ea75b28
commit e82505d20e
9 changed files with 82 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/maxpool_layer.c
View File

@ -29,7 +29,7 @@ void cudnn_maxpool_setup(layer *l)
maxpool_status = cudnnSetPooling2dDescriptor(
l->poolingDesc,
CUDNN_POOLING_MAX,
CUDNN_PROPAGATE_NAN, // CUDNN_PROPAGATE_NAN, CUDNN_NOT_PROPAGATE_NAN
CUDNN_NOT_PROPAGATE_NAN, // CUDNN_PROPAGATE_NAN, CUDNN_NOT_PROPAGATE_NAN
l->size,
l->size,
0, //l.pad,