未验证 提交 10b3cbfb 编写于 作者: Y yuyang18

Fix CI hang

上级 50eaa027
...@@ -56,7 +56,8 @@ def simple_fc_net(in_size, ...@@ -56,7 +56,8 @@ def simple_fc_net(in_size,
capacity=queue_capacity, capacity=queue_capacity,
shapes=[[-1, in_size], [-1, 1]], shapes=[[-1, in_size], [-1, 1]],
lod_levels=[0, 0], lod_levels=[0, 0],
dtypes=['float32', 'int64']) dtypes=['float32', 'int64'],
use_double_buffer=False)
feed_queue = reader.queue feed_queue = reader.queue
reader = fluid.layers.batch(reader, batch_size=batch_size) reader = fluid.layers.batch(reader, batch_size=batch_size)
if use_double_buffer: if use_double_buffer:
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册