Fixed gradient_centralization for Grouped-Conv

5 years ago · 74134e5389
parent 9f8b1e31a8
commit 74134e5389
1 changed files with 1 additions and 1 deletions
--- a/src/convolutional_kernels.cu
+++ b/src/convolutional_kernels.cu
@ -1252,7 +1252,7 @@ void update_convolutional_layer_gpu(layer l, int batch, float learning_rate_init
    if (l.grad_centr && l.batch_normalize) {
        // weights[filters][channels][height][width]
        // for(filters) w[f] = w[f] - mean(w[c][h][w])
-        gradient_centralization_gpu(l.size, l.size, l.c, l.n, l.weight_updates_gpu);
+        gradient_centralization_gpu(l.size, l.size, l.c / l.groups, l.n, l.weight_updates_gpu);
    }