Changes to make routing work better

This commit is contained in:
Joseph Redmon
2015-07-21 16:09:33 -07:00
parent e56d1eff13
commit d00f0a1ccd
18 changed files with 29 additions and 86 deletions

View File

@ -35,7 +35,7 @@ __global__ void backward_avgpool_layer_kernel(int n, int w, int h, int c, float
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);
}
}