mailcorahul:

class BinaryMNIST(Dataset):
    def __init__(self):
        self.images , self.labels = ##using datasets.MNIST
    def __getitem__(self, idx):
        image, label = self.images[idx], self.labels[idx]
        image[image < 127] = 0
        image[image >= 127] = 255
        # do the necessary transforms ...
        return image, label



hello @mailcorahul
do you think normalizing would make sense if I binarize the MNIST set?
thanks in advance!