Unverified Commit ed346d82 authored by djqualia's avatar djqualia Committed by GitHub
Browse files

Update data.py 

Fix edge case where data.size == max_length (resulting in torch.randint(0,0,...) which produces an error.
parent 6bf92afa

audiolm_pytorch/data.py

+1 −1
Original line number Diff line number Diff line
@@ -52,7 +52,7 @@ class SoundDataset(Dataset): 


        data, sample_hz = torchaudio.load(file)

        

        if data.size(1) >= self.max_length:

        if data.size(1) > self.max_length:

            max_start = data.size(1) - self.max_length

            start = torch.randint(0, max_start, (1, ))

            data = data[:, start:start + self.max_length]