Unverified Commit 2f35f828 authored by Phil Wang's avatar Phil Wang Committed by GitHub
Browse files

Merge pull request #28 from djqualia/main 

Update data.py
parents 6bf92afa ed346d82

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]