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

More fixes

Browse Source
This commit is contained in:
James Betker 2021-09-14 20:45:05 -06:00
parent 9a9c90660f
commit 8d9857f33d
1 changed files with 7 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -84,12 +84,12 @@ class UnsupervisedAudioDataset(torch.utils.data.Dataset):
assert audiopath in related_files
assert len(related_files) < 50000 # Sanity check to ensure we aren't loading "related files" that aren't actually related.
if len(related_files) == 0:
j = 0
print(f"No related files for {audiopath}")
related_files.remove(audiopath)
related_clips = []
random.shuffle(related_clips)
for j, related_file in enumerate(related_files):
i = 0
for related_file in related_files:
rel_clip = load_audio(related_file, self.sampling_rate)
gap = rel_clip.shape[-1] - self.extra_sample_len
if gap < 0:
@ -98,12 +98,13 @@ class UnsupervisedAudioDataset(torch.utils.data.Dataset):
rand_start = random.randint(0, gap)
rel_clip = rel_clip[:, rand_start:rand_start+self.extra_sample_len]
related_clips.append(rel_clip)
if j >= self.extra_samples:
i += 1
if i >= self.extra_samples:
break
actual_extra_samples = j
while j < self.extra_samples:
actual_extra_samples = i
while i < self.extra_samples:
related_clips.append(torch.zeros(1, self.extra_sample_len))
j += 1
i += 1
return torch.stack(related_clips, dim=0), actual_extra_samples
def __getitem__(self, index):