From 20312211e0ad30cdbbbf3fd4cb421c1071f2b9d7 Mon Sep 17 00:00:00 2001 From: James Betker Date: Thu, 20 Jan 2022 11:28:12 -0700 Subject: [PATCH] Fix bug in code alignment --- codes/data/audio/fast_paired_dataset.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/codes/data/audio/fast_paired_dataset.py b/codes/data/audio/fast_paired_dataset.py index 074cd647..34a62ebd 100644 --- a/codes/data/audio/fast_paired_dataset.py +++ b/codes/data/audio/fast_paired_dataset.py @@ -155,7 +155,7 @@ class FastPairedVoiceDataset(torch.utils.data.Dataset): 'real_text': text, 'padded_text': tseq, '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), 'wav': wav, 'wav_lengths': torch.tensor(orig_output, dtype=torch.long),