From fc656455f63a6d45220ab228fa5eb4bc172708d5 Mon Sep 17 00:00:00 2001
From: yongzhuo <31341349+yongzhuo@users.noreply.github.com>
Date: Sun, 16 Jun 2019 19:39:17 +0800
Subject: [PATCH] Update args.py

---
 ClassificationText/bert/args.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/ClassificationText/bert/args.py b/ClassificationText/bert/args.py
index ec7bc4f..d98ac71 100644
--- a/ClassificationText/bert/args.py
+++ b/ClassificationText/bert/args.py
@@ -16,7 +16,7 @@ label = 2
 l2 = 0.32
 
 epochs = 100
-batch_size = 128
+batch_size = 2
 path_save_model = 'model_webank_tdt/bert_avt_cnn.h5' # 'bert_bi_lstm_pair.h5'
 
 # text-cnn