mrq/DL-Art-School
1
3
Fork 5
Code Issues Pull Requests Packages Projects Releases Wiki Activity

unsupervised_audio_dataset: try to recover from failures of audio2numpy

Browse Source
This commit is contained in:
James Betker 2021-09-17 15:25:57 -06:00
parent 5c8d266d4f
commit 359e9e27a7
1 changed files with 7 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -108,9 +108,13 @@ class UnsupervisedAudioDataset(torch.utils.data.Dataset):
return torch.stack(related_clips, dim=0), actual_extra_samples
def __getitem__(self, index):
# Split audio_norm into two tensors of equal size.
audio_norm, filename = self.get_audio_for_index(index)
alt_files, actual_samples = self.get_related_audio_for_index(index)
try:
# Split audio_norm into two tensors of equal size.
audio_norm, filename = self.get_audio_for_index(index)
alt_files, actual_samples = self.get_related_audio_for_index(index)
except:
print(f"Error loading audio for file {filename} or {alt_files}")
return self[index+1]
# This is required when training to make sure all clips align.
if self.pad_to is not None: