Mirrors/TTS
1
0
Fork 0
mirror of https://github.com/coqui-ai/TTS.git synced 2026-02-24 20:19:54 +01:00
Code Issues Packages Projects Releases Wiki Activity

Style fixed

Browse Source
This commit is contained in:
Daniel Vera Nieto
2023-03-13 16:11:15 +01:00
parent 0d12229b64
commit dfb48737fb
1 changed files with 7 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -1630,13 +1630,13 @@ class Vits(BaseTTS):
else:
if num_gpus > 1:
loader = DataLoader(
dataset,
sampler=sampler,
batch_size=config.eval_batch_size if is_eval else config.batch_size,
collate_fn=dataset.collate_fn,
num_workers=config.num_eval_loader_workers if is_eval else config.num_loader_workers,
pin_memory=False,
)
dataset,
sampler=sampler,
batch_size=config.eval_batch_size if is_eval else config.batch_size,
collate_fn=dataset.collate_fn,
num_workers=config.num_eval_loader_workers if is_eval else config.num_loader_workers,
pin_memory=False,
)
else:
loader = DataLoader(
dataset,