deviandice/DL-Art-School
1
0
Fork 0
forked from mrq/DL-Art-School
Code Issues Pull Requests Packages Projects Releases Wiki Activity

paired fix

Browse Source
This commit is contained in:
James Betker 2022-04-16 23:41:57 -06:00
parent 8fe0dff33c
commit c85ab738c5
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
codes/data/audio/paired_voice_audio_dataset.py
View File

@ -207,7 +207,7 @@ class TextWavLoader(torch.utils.data.Dataset):
return self[(index+1) % len(self)] return self[(index+1) % len(self)]
if self.load_aligned_codes: if self.load_aligned_codes:
aligned_codes = self.audiopaths_and_text[index][2] aligned_codes = self.audiopaths_and_text[index][3]
actually_skipped_items = self.skipped_items actually_skipped_items = self.skipped_items
self.skipped_items = 0 self.skipped_items = 0
@ -285,8 +285,8 @@ if __name__ == '__main__':
batch_sz = 8 batch_sz = 8
params = { params = {
'mode': 'paired_voice_audio', 'mode': 'paired_voice_audio',
'path': ['Y:\\clips\\books1\\transcribed-oco.tsv'], 'path': ['Y:\\libritts/test-clean_list.txt'],
'fetcher_mode': ['tsv'], 'fetcher_mode': ['libritts'],
'phase': 'train', 'phase': 'train',
'n_workers': 0, 'n_workers': 0,
'batch_size': batch_sz, 'batch_size': batch_sz,
@ -297,7 +297,7 @@ if __name__ == '__main__':
'num_conditioning_candidates': 2, 'num_conditioning_candidates': 2,
'conditioning_length': 44000, 'conditioning_length': 44000,
'use_bpe_tokenizer': True, 'use_bpe_tokenizer': True,
'load_aligned_codes': True, 'load_aligned_codes': False,
} }
from data import create_dataset, create_dataloader from data import create_dataset, create_dataloader