From 231607a546f6c8a85a9710b2b48dc04fadda806f Mon Sep 17 00:00:00 2001 From: Eren Date: Wed, 26 Sep 2018 16:52:47 +0200 Subject: [PATCH] Reduce batch group size --- train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/train.py b/train.py index 1cdca458..ffe2557e 100644 --- a/train.py +++ b/train.py @@ -366,7 +366,7 @@ def main(args): c.r, c.text_cleaner, ap=ap, - batch_group_size=16*c.batch_size, + batch_group_size=8*c.batch_size, min_seq_len=c.min_seq_len) train_loader = DataLoader(