produce correct clip_lengths..

This commit is contained in:
James Betker 2022-06-21 20:21:12 -06:00
parent 4a1f3aba31
commit f0117150d0

View File

@ -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: