fix tensor lengths

2022-01-01 00:23:46 -07:00 · 2022-01-01 00:23:46 -07:00 · c28d8770c7
commit c28d8770c7
parent bbacffb790
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
@ -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: