produce correct clip_lengths..

2022-06-21 20:21:12 -06:00 · 2022-06-21 20:21:12 -06:00 · f0117150d0
commit f0117150d0
parent 4a1f3aba31
1 changed files with 1 additions and 1 deletions
--- a/codes/data/audio/unsupervised_audio_dataset.py
+++ b/codes/data/audio/unsupervised_audio_dataset.py
@ -168,7 +168,7 @@ class UnsupervisedAudioDataset(torch.utils.data.Dataset):
        output = {
            'prepad_length': prepad_length,
            'clip': clips[0],
-            'clip_lengths': torch.tensor(audio_norm.shape[-1]),
+            'clip_lengths': torch.tensor(clips[0].shape[-1]),
            'path': filename,
        }
        if self.should_resample_clip: