mrq/vall-e
5
9
Fork 0
Code Issues 8 Pull Requests Packages Projects Releases Wiki Activity

ugh

Browse Source
This commit is contained in:
mrq 2024-06-01 10:30:13 -05:00
parent 39bc019142
commit d0ebce6bac
2 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
vall_e/data.py
View File

@ -242,8 +242,8 @@ class Dataset(_Dataset):
# recreate paths_by_spkr_name # recreate paths_by_spkr_name
self.paths_by_spkr_name = {} self.paths_by_spkr_name = {}
for path in self.paths: for path in self.paths:
name = cfg.get_spkr( path ) name = cfg.get_spkr( Path(path) )
if name not in self.paths_by_spkr_name[name]: if name not in self.paths_by_spkr_name:
self.paths_by_spkr_name[name] = [] self.paths_by_spkr_name[name] = []
self.paths_by_spkr_name[name].append( path ) self.paths_by_spkr_name[name].append( path )

2
vall_e/utils/trainer.py
View File

@ -150,8 +150,10 @@ def train(
last_save_step = engines.global_step last_save_step = engines.global_step
last_eval_step = 0 last_eval_step = 0
"""
if cfg.distributed: if cfg.distributed:
train_dl.sampler.set_epoch(int(engines.global_samples / len(train_dl.dataset.paths))) train_dl.sampler.set_epoch(int(engines.global_samples / len(train_dl.dataset.paths)))
"""
# Training loop # Training loop
for batch in _make_infinite_epochs(train_dl): for batch in _make_infinite_epochs(train_dl):