From 3f03e3012c22e8f86a7e9c0b6d52ccde36d1b15e Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Eren=20G=C3=B6lge?= Date: Sat, 7 May 2022 13:40:34 +0200 Subject: [PATCH] Fix batch_group_size in VITS --- TTS/tts/models/vits.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/TTS/tts/models/vits.py b/TTS/tts/models/vits.py index 613e4eae..3bcd59a1 100644 --- a/TTS/tts/models/vits.py +++ b/TTS/tts/models/vits.py @@ -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,