blakelemoine/DL-Art-School
1
0
Fork 0
forked from mrq/DL-Art-School
Code Pull Requests Activity

produce correct clip_lengths..

Browse Source
This commit is contained in:
James Betker 2022-06-21 20:21:12 -06:00
parent 4a1f3aba31
commit f0117150d0
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
codes/data/audio/unsupervised_audio_dataset.py
View File

@ -168,7 +168,7 @@ class UnsupervisedAudioDataset(torch.utils.data.Dataset):
output = { output = {
'prepad_length': prepad_length, 'prepad_length': prepad_length,
'clip': clips[0], 'clip': clips[0],
'clip_lengths': torch.tensor(audio_norm.shape[-1]), 'clip_lengths': torch.tensor(clips[0].shape[-1]),
'path': filename, 'path': filename,
} }
if self.should_resample_clip: if self.should_resample_clip: