From 359e9e27a7c989a86af65b2dfd5765b2ccb2041e Mon Sep 17 00:00:00 2001 From: James Betker Date: Fri, 17 Sep 2021 15:25:57 -0600 Subject: [PATCH] unsupervised_audio_dataset: try to recover from failures of audio2numpy --- codes/data/audio/unsupervised_audio_dataset.py | 10 +++++++--- 1 file changed, 7 insertions(+), 3 deletions(-) diff --git a/codes/data/audio/unsupervised_audio_dataset.py b/codes/data/audio/unsupervised_audio_dataset.py index c1a38eaf..5e4fd300 100644 --- a/codes/data/audio/unsupervised_audio_dataset.py +++ b/codes/data/audio/unsupervised_audio_dataset.py @@ -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: