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

fix tensor lengths

Browse Source
This commit is contained in:
James Betker 2022-01-01 00:23:46 -07:00
parent bbacffb790
commit c28d8770c7
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

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