From c28d8770c7fd2a6b73e2fbc33fee175db98b5cf9 Mon Sep 17 00:00:00 2001 From: James Betker Date: Sat, 1 Jan 2022 00:23:46 -0700 Subject: [PATCH] fix tensor 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 11350429..e82d857e 100644 --- a/codes/data/audio/unsupervised_audio_dataset.py +++ b/codes/data/audio/unsupervised_audio_dataset.py @@ -162,7 +162,7 @@ class UnsupervisedAudioDataset(torch.utils.data.Dataset): output = { 'clip': clips[0], - 'clip_lengths': audio_norm.shape[-1], + 'clip_lengths': torch.tensor(audio_norm.shape[-1]), 'path': filename, } if self.should_resample_clip: