src/convolutional_kernels.cu
@@ -133,6 +133,9 @@ if(l.batch_normalize){ backward_batchnorm_layer_gpu(l, state); //axpy_ongpu(l.outputs*l.batch, -state.net.decay, l.x_gpu, 1, l.delta_gpu, 1); } else { //axpy_ongpu(l.outputs*l.batch, -state.net.decay, l.output_gpu, 1, l.delta_gpu, 1); } float *original_input = state.input;