From 4d059ccf8dcef617b5ffacf7e2b60121d37a164d Mon Sep 17 00:00:00 2001
From: nttstar <guojia@gmail.com>
Date: Fri, 15 Feb 2019 14:59:09 +0800
Subject: [PATCH] tiny fix

---
 recognition/metric.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/recognition/metric.py b/recognition/metric.py
index cf8afca..0758c52 100644
--- a/recognition/metric.py
+++ b/recognition/metric.py
@@ -13,7 +13,7 @@ class AccMetric(mx.metric.EvalMetric):
   def update(self, labels, preds):
     self.count+=1
     label = labels[0]
-    pred_label = preds[0]
+    pred_label = preds[1]
     #print('ACC', label.shape, pred_label.shape)
     if pred_label.shape != label.shape:
         pred_label = mx.ndarray.argmax(pred_label, axis=self.axis)