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

oops

Browse Source
This commit is contained in:
mrq 2023-10-06 09:26:52 -05:00
parent 2f2505b12f
commit 82f02ae9b1
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
vall_e/data.py
View File

@ -115,7 +115,7 @@ def _get_hdf5_paths( data_dir, type="training", validate=False ):
duration = child.attrs['duration'] duration = child.attrs['duration']
if type not in _total_durations: if type not in _total_durations:
_total_durations[type] = 0 _total_durations[type] = 0
_total_durations[type] += entry['duration'] _total_durations[type] += child.attrs['duration']
return cfg.dataset.min_duration <= duration and duration <= cfg.dataset.max_duration and cfg.dataset.min_phones <= phones and phones <= cfg.dataset.max_phones return cfg.dataset.min_duration <= duration and duration <= cfg.dataset.max_duration and cfg.dataset.min_phones <= phones and phones <= cfg.dataset.max_phones
key = f"/{type}{_get_hdf5_path(data_dir)}" key = f"/{type}{_get_hdf5_path(data_dir)}"