提交 a6364f9e 编写于 作者: Q qiaolongfei

change batch_size to 5

上级 e60d94b3
...@@ -93,12 +93,12 @@ def main(): ...@@ -93,12 +93,12 @@ def main():
trn_reader = paddle.reader.batched( trn_reader = paddle.reader.batched(
paddle.reader.shuffle( paddle.reader.shuffle(
train_reader("data/pre-wmt14/train/train"), buf_size=8192), train_reader("data/pre-wmt14/train/train"), buf_size=8192),
batch_size=10000) batch_size=5)
trainer.train( trainer.train(
reader=trn_reader, reader=trn_reader,
event_handler=event_handler, event_handler=event_handler,
num_passes=10, num_passes=10000,
reader_dict=reader_dict) reader_dict=reader_dict)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册