diff --git a/demo/seqToseq/api_train_v2.py b/demo/seqToseq/api_train_v2.py index bdcf3a5af074d2c91f121be2610d63ba9c5009e8..74ae1cf9ec81b00b55202d7c1316c7cb074cfd43 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)