提交 d27a3f91 编写于 作者: R ranqiu

Fix bugs of generate_sequence_by_rnn_lm

上级 7631f3b4
...@@ -109,7 +109,7 @@ def main(): ...@@ -109,7 +109,7 @@ def main():
test_reader = paddle.batch( test_reader = paddle.batch(
paddle.reader.shuffle( paddle.reader.shuffle(
reader.rnn_reader(**reader_args), buf_size=65536), reader.rnn_reader(**reader_args), buf_size=65536),
batch_size=config.batch_size) batch_size=conf.batch_size)
train( train(
topology=cost, topology=cost,
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册