未验证 提交 ade6c832 编写于 作者: Q QI JUN 提交者: GitHub

open test_word2vec (#6104)

上级 42708ded
...@@ -58,10 +58,6 @@ train_reader = paddle.batch( ...@@ -58,10 +58,6 @@ train_reader = paddle.batch(
place = fluid.CPUPlace() place = fluid.CPUPlace()
exe = fluid.Executor(place) exe = fluid.Executor(place)
# fix https://github.com/PaddlePaddle/Paddle/issues/5434 then remove
# below exit line.
exit(0)
exe.run(fluid.default_startup_program()) exe.run(fluid.default_startup_program())
for pass_id in range(PASS_NUM): for pass_id in range(PASS_NUM):
...@@ -79,6 +75,6 @@ for pass_id in range(PASS_NUM): ...@@ -79,6 +75,6 @@ for pass_id in range(PASS_NUM):
'nextw': input_data[4] 'nextw': input_data[4]
}, },
fetch_list=[avg_cost]) fetch_list=[avg_cost])
if avg_cost_np[0] < 10.0: if avg_cost_np[0] < 5.0:
exit(0) # if avg cost less than 10.0, we think our code is good. exit(0) # if avg cost less than 10.0, we think our code is good.
exit(1) exit(1)
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册