forked from mrq/DL-Art-School
produce correct clip_lengths..
This commit is contained in:
parent
4a1f3aba31
commit
f0117150d0
|
@ -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:
|
||||
|
|
Loading…
Reference in New Issue
Block a user