Joseph Redmon
2015-08-25 9d42f49a240136a8cd643cdc1f98230d4f22b05e
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);
    }
}