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

Fix bug in code alignment

Browse Source
This commit is contained in:
James Betker 2022-01-20 11:28:12 -07:00
parent 8e2439f50d
commit 20312211e0
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
codes/data/audio/fast_paired_dataset.py
View File

@ -155,7 +155,7 @@ class FastPairedVoiceDataset(torch.utils.data.Dataset):
'real_text': text, 'real_text': text,
'padded_text': tseq, 'padded_text': tseq,
'aligned_codes': aligned_codes, 'aligned_codes': aligned_codes,
'aligned_code_lengths': orig_aligned_code_length, 'aligned_codes_lengths': orig_aligned_code_length,
'text_lengths': torch.tensor(orig_text_len, dtype=torch.long), 'text_lengths': torch.tensor(orig_text_len, dtype=torch.long),
'wav': wav, 'wav': wav,
'wav_lengths': torch.tensor(orig_output, dtype=torch.long), 'wav_lengths': torch.tensor(orig_output, dtype=torch.long),