Medix-AI/coqui-tts
Medix-AI
/
coqui-tts
mirror of https://github.com/coqui-ai/TTS.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

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

This commit is contained in:
Jindrich Matousek 2023-09-06 13:59:24 +02:00
parent 0938f1cfa1
commit a0db2eeee8
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
TTS/tts/models/vits.py
View File

@ -1679,7 +1679,8 @@ class Vits(BaseTTS):
dataset.preprocess_samples() dataset.preprocess_samples()
# get samplers # 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: if sampler is None:
loader = DataLoader( loader = DataLoader(
dataset, dataset,