Update train.py (#1590)

4af8f461 · kolinwei · Yibing Liu · 4a676b02 · 4af8f461
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

fluid/PaddleNLP/text_classification/train.py fluid/PaddleNLP/text_classification/train.py +1 -1

未找到文件。
--- a/fluid/PaddleNLP/text_classification/train.py
+++ b/fluid/PaddleNLP/text_classification/train.py
@@ -89,7 +89,7 @@ def train(train_reader,
 def train_net():
    word_dict, train_reader, test_reader = utils.prepare_data(
-        "imdb", self_dict=False, batch_size=4, buf_size=50000)
+        "imdb", self_dict=False, batch_size=128, buf_size=50000)
    if sys.argv[1] == "bow":
        train(