Seich
/
mayavoz
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Merge branch 'dev' of https://github.com/shahules786/enhancer into dev-hawk

This commit is contained in:
shahules786 2022-10-27 16:18:53 +05:30
parent 430696cfac fb2543e81e
commit 3a9e577ccb
1 changed files with 1 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
enhancer/data/dataset.py
View File

@ -124,10 +124,7 @@ class TaskDataset(pl.LightningDataModule):
valid_sec_now = 0.0 valid_sec_now = 0.0
valid_indices = [] valid_indices = []
all_speakers = np.unique( all_speakers = np.unique(
[ [Path(file["clean"]).name.split("_")[0] for file in data]
(Path(file["clean"]).name.split("_")[0], file["duration"])
for file in data
]
) )
possible_indices = list(range(0, len(all_speakers))) possible_indices = list(range(0, len(all_speakers)))
rng = create_unique_rng(len(all_speakers)) rng = create_unique_rng(len(all_speakers))