proper flatten in acc (apache#9619)

jinhuang415 · Jan 29, 2018 · ed823b2 · ed823b2
1 parent 5a5cd64
commit ed823b2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/mxnet/metric.py b/python/mxnet/metric.py
@@ -399,7 +399,7 @@ def update(self, labels, preds):
             if pred_label.context != label.context:
                 pred_label = pred_label.as_in_context(label.context)
 
-            self.sum_metric += (pred_label.flatten() == label.flatten()).sum().asscalar()
+            self.sum_metric += (pred_label.reshape((-1,)) == label.reshape((-1,))).sum().asscalar()
             self.num_inst += numpy.prod(pred_label.shape)