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

hate warnings

Browse Source
This commit is contained in:
Joseph Redmon
2016-06-14 11:30:28 -07:00
parent 729ce43e6e
commit 8322a58cf6
9 changed files with 89 additions and 94 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
src/network.c
View File

@ -392,6 +392,11 @@ void set_batch_network(network *net, int b)
int i;
for(i = 0; i < net->n; ++i){
net->layers[i].batch = b;
#ifdef CUDNN
if(net->layers[i].type == CONVOLUTIONAL){
cudnn_convolutional_setup(net->layers + i);
}
#endif
}
}