Joseph Redmon
2016-06-14 8322a58cf69e79ff1847634ede1a2a1b2e71e6d6
src/network.c
@@ -392,6 +392,11 @@
    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
    }
}