Fix batch_group_size in VITS

2026-09-09 11:16:00 +08:00 · 2022-05-07 13:44:44 +02:00
parent fa887ef5f9
commit 3f03e3012c
1 changed files with 1 additions and 1 deletions
@@ -1452,7 +1452,7 @@ class Vits(BaseTTS):
            # init dataloader
            dataset = VitsDataset(
                samples=samples,
-                # batch_group_size=0 if is_eval else config.batch_group_size * config.batch_size,
+                batch_group_size=0 if is_eval else config.batch_group_size * config.batch_size,
                min_text_len=config.min_text_len,
                max_text_len=config.max_text_len,
                min_audio_len=config.min_audio_len,