wassname/TTS
1
0
Fork 0
mirror of https://github.com/wassname/TTS.git synced 2026-09-09 11:16:00 +08:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #2407 from dveni/patch-1

Browse Source 
Update vits.py
This commit is contained in:
Eren Gölge
2023-03-14 10:28:23 +01:00
committed by GitHub
parent c10f9a3699 dfb48737fb
commit b8d9837d27
1 changed files with 17 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files
TTS/tts/models/vits.py
+17 -7
View File
@@ -1628,13 +1628,23 @@ class Vits(BaseTTS):
pin_memory=False,
)
else:
loader = DataLoader(
dataset,
batch_sampler=sampler,
collate_fn=dataset.collate_fn,
num_workers=config.num_eval_loader_workers if is_eval else config.num_loader_workers,
pin_memory=False,
)
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,
)
else:
loader = DataLoader(
dataset,
batch_sampler=sampler,
collate_fn=dataset.collate_fn,
num_workers=config.num_eval_loader_workers if is_eval else config.num_loader_workers,
pin_memory=False,
)
return loader
def get_optimizer(self) -> List: