Joseph Redmon
2016-06-09 32d2c969973aa98635123743f321859192ff581d
src/convolutional_kernels.cu
@@ -161,6 +161,7 @@
            l.filter_updates_gpu);
    if(state.delta){
        if(l.binary || l.xnor) swap_binary(&l);
        cudnnConvolutionBackwardData(cudnn_handle(),
                &one,
                l.filterDesc,
@@ -174,6 +175,7 @@
                &one,
                l.dsrcTensorDesc,
                state.delta);
        if(l.binary || l.xnor) swap_binary(&l);
    }
#else