From adc66c04a16fff735e1c7374beb940b16d915a40 Mon Sep 17 00:00:00 2001
From: tianxin04 <tianxin04@baidu.com>
Date: Mon, 17 Jun 2019 11:43:22 +0800
Subject: [PATCH] fix #138

---
 ERNIE/finetune/classifier.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/ERNIE/finetune/classifier.py b/ERNIE/finetune/classifier.py
index 48139c8..950c550 100644
--- a/ERNIE/finetune/classifier.py
+++ b/ERNIE/finetune/classifier.py
@@ -156,7 +156,7 @@ def evaluate(exe, test_program, test_pyreader, graph_vars, eval_phase):
         outputs = exe.run(fetch_list=train_fetch_list)
         ret = {"loss": np.mean(outputs[0]), "accuracy": np.mean(outputs[1])}
         if "learning_rate" in graph_vars:
-            ret["learning_rate"] = float(outputs[4][0])
+            ret["learning_rate"] = float(outputs[3][0])
         return ret
 
     test_pyreader.start()
-- 
GitLab