mirror of
https://github.com/pjreddie/darknet.git
synced 2023-08-10 21:13:14 +03:00
Changes to make routing work better
This commit is contained in:
@ -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);
|
||||
}
|
||||
}
|
||||
|
||||
|
Reference in New Issue
Block a user