From a6364f9e6a84f3d29876c5ad286acc1e86f7fe1f Mon Sep 17 00:00:00 2001 From: qiaolongfei Date: Fri, 3 Mar 2017 16:27:29 +0800 Subject: [PATCH] change batch_size to 5 --- demo/seqToseq/api_train_v2.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/demo/seqToseq/api_train_v2.py b/demo/seqToseq/api_train_v2.py index bdcf3a5af..74ae1cf9e 100644 --- a/demo/seqToseq/api_train_v2.py +++ b/demo/seqToseq/api_train_v2.py @@ -93,12 +93,12 @@ def main(): trn_reader = paddle.reader.batched( paddle.reader.shuffle( train_reader("data/pre-wmt14/train/train"), buf_size=8192), - batch_size=10000) + batch_size=5) trainer.train( reader=trn_reader, event_handler=event_handler, - num_passes=10, + num_passes=10000, reader_dict=reader_dict) -- GitLab