Reduce batch group size

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