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 #2191

This commit is contained in:
Eren G??lge 2023-01-15 23:11:57 +01:00
parent bc422f2f3c
commit 6e3f74fc29
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
TTS/tts/models/base_tts.py
View File

@ -345,7 +345,7 @@ class BaseTTS(BaseTrainerModel):
loader = DataLoader( loader = DataLoader(
dataset, dataset,
batch_size=config.eval_batch_size if is_eval else config.batch_size, batch_size=config.eval_batch_size if is_eval else config.batch_size,
shuffle=config.shuffle if sampler is not None else False, # if there is no other sampler shuffle=config.shuffle if sampler is None else False, # if there is no other sampler
collate_fn=dataset.collate_fn, collate_fn=dataset.collate_fn,
drop_last=config.drop_last, # setting this False might cause issues in AMP training. drop_last=config.drop_last, # setting this False might cause issues in AMP training.
sampler=sampler, sampler=sampler,