未验证 提交 b2be1faa 编写于 作者: T tianxin 提交者: GitHub

Merge pull request #164 from tianxin1860/develop

fix #138
...@@ -156,7 +156,7 @@ def evaluate(exe, test_program, test_pyreader, graph_vars, eval_phase): ...@@ -156,7 +156,7 @@ def evaluate(exe, test_program, test_pyreader, graph_vars, eval_phase):
outputs = exe.run(fetch_list=train_fetch_list) outputs = exe.run(fetch_list=train_fetch_list)
ret = {"loss": np.mean(outputs[0]), "accuracy": np.mean(outputs[1])} ret = {"loss": np.mean(outputs[0]), "accuracy": np.mean(outputs[1])}
if "learning_rate" in graph_vars: if "learning_rate" in graph_vars:
ret["learning_rate"] = float(outputs[4][0]) ret["learning_rate"] = float(outputs[3][0])
return ret return ret
test_pyreader.start() test_pyreader.start()
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册