From 4af8f461ef299582aff61b6c213cf9371b6a853f Mon Sep 17 00:00:00 2001
From: kolinwei <331911734@qq.com>
Date: Thu, 3 Jan 2019 11:06:46 +0800
Subject: [PATCH] Update train.py (#1590)

---
 fluid/PaddleNLP/text_classification/train.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/fluid/PaddleNLP/text_classification/train.py b/fluid/PaddleNLP/text_classification/train.py
index 159266f3..174636f0 100644
--- 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(
-- 
GitLab