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

3
src/network.c
View File

@ -377,6 +377,9 @@ void set_batch_network(network *net, int b)
}
*/
}
else if (net->layers[i].type == MAXPOOL) {
cudnn_maxpool_setup(net->layers + i);
}
#endif
}
}