src/avgpool_layer_kernels.cu @@ -35,7 +35,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); } }