fix fpgm prune criterion bug test=develop (PaddlePaddle#234)

lizexu123 · Apr 23, 2020 · 5b3bd16 · 5b3bd16
1 parent a13fe40
commit 5b3bd16
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/paddleslim/prune/criterion.py b/paddleslim/prune/criterion.py
@@ -63,7 +63,7 @@ def geometry_median(group, graph):
     x = w.repeat(channel_num, axis=0)
     y = np.zeros_like(x)
     for i in range(channel_num):
-        y[i * channel_num:(i + 1) * channel_num] = np.tile(channel_num,
+        y[i * channel_num:(i + 1) * channel_num] = np.tile(w[i],
                                                            (channel_num, 1))
     tmp = np.sqrt(np.sum((x - y)**2, -1))
     tmp = tmp.reshape((channel_num, channel_num))