Fix: add `is_eval` when calling `get_sampler` to respect training/validation

2023-09-06 13:59:24 +02:00 · 2023-09-06 13:59:24 +02:00 · a0db2eeee8
parent 0938f1cfa1
commit a0db2eeee8
1 changed files with 2 additions and 1 deletions
--- a/TTS/tts/models/vits.py
+++ b/TTS/tts/models/vits.py
@ -1679,7 +1679,8 @@ class Vits(BaseTTS):
            dataset.preprocess_samples()

            # get samplers
-            sampler = self.get_sampler(config, dataset, num_gpus)
+            # JMa: Add `is_eval` parameter because the default is `False` and `batch_size` was used instead of `eval_batch_size`
+            sampler = self.get_sampler(config, dataset, num_gpus, is_eval)
            if sampler is None:
                loader = DataLoader(
                    dataset,