AlexeyAB
2017-01-15 e83d8e56d56978e5c2e571b8b4bf05b90a851ce0
src/convolutional_kernels.cu
@@ -2,6 +2,8 @@
#include "curand.h"
#include "cublas_v2.h"
#pragma comment(lib, "cudnn.lib")
extern "C" {
#include "convolutional_layer.h"
#include "batchnorm_layer.h"
@@ -133,6 +135,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;