From f0117150d0061cd2cf07b1c3411e4715ba497229 Mon Sep 17 00:00:00 2001 From: James Betker Date: Tue, 21 Jun 2022 20:21:12 -0600 Subject: [PATCH] produce correct clip_lengths.. --- codes/data/audio/unsupervised_audio_dataset.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/codes/data/audio/unsupervised_audio_dataset.py b/codes/data/audio/unsupervised_audio_dataset.py index ce73c440..5118329a 100644 --- 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: