src/avgpool_layer_kernels.cu
@@ -1,3 +1,7 @@ #include "cuda_runtime.h" #include "curand.h" #include "cublas_v2.h" extern "C" { #include "avgpool_layer.h" #include "cuda.h" @@ -35,7 +39,7 @@ int out_index = (k + c*b); for(i = 0; i < w*h; ++i){ int in_index = i + h*w*(k + b*c); in_delta[in_index] = out_delta[out_index] / (w*h); in_delta[in_index] += out_delta[out_index] / (w*h); } }