From a0db2eeee81afa64657a1268b867fa02b0035c11 Mon Sep 17 00:00:00 2001
From: Jindrich Matousek <jindrich.matousek@gmail.com>
Date: Wed, 6 Sep 2023 13:59:24 +0200
Subject: [PATCH] Fix: add `is_eval` when calling `get_sampler` to respect
 training/validation

---
 TTS/tts/models/vits.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/TTS/tts/models/vits.py b/TTS/tts/models/vits.py
index 057ffe7e..c11d9b96 100644
--- 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,