another fix

This commit is contained in:
James Betker 2021-08-31 14:46:46 -06:00
parent 8810d3de97
commit 30cd33fe44

View File

@ -92,10 +92,6 @@ class WavfileDataset(torch.utils.data.Dataset):
start = random.randint(0, gap-1)
audio_norm = audio_norm[:, start:start+self.pad_to]
# Bail and try the next clip if there is not enough data.
if audio_norm.shape[-1] < self.min_sz:
return self[(index + 1) % len(self)]
output = {
'clip': audio_norm,
'path': filename,