Reduce batch group size

2018-09-26 16:52:47 +02:00 · 2018-09-26 16:52:47 +02:00 · 231607a546
parent 006354320e
commit 231607a546
1 changed files with 1 additions and 1 deletions
--- 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(