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

Merge pull request #34 from djqualia/patch-2 

Pad right side of audio vs. left
parents b0384e83 4b893421

audiolm_pytorch/data.py

+1 −1
Original line number Diff line number Diff line
@@ -61,7 +61,7 @@ class SoundDataset(Dataset): 
            data = data[:, start:start + self.max_length]



        else:

            data = torch.nn.functional.pad(data, (self.max_length - data.size(1), 0), 'constant')

            data = torch.nn.functional.pad(data, (0, self.max_length - data.size(1)), 'constant')

        

        data = rearrange(data, '1 ... -> ...')