dataset.py

class NoiseDataSet(Dataset):
    def __init__(self, X, y):
        super().__init__()
        assert(len(X) == len(y))

        self.X = X
        self.y = y

    def __len__(self):
        return len(self.X)

    def __getitem__(self, idx):
        filename = self.X[idx]
        wav = wavfile_to_examples(filename)
        wav = wav.cuda()

        target = self.y[idx]

        return wav, target