Method transforms

datamodule.py:231–237

(self)

        pass

    def transforms(self):
        tform = transforms.Compose([
            transforms.Resize(self.size),
            transforms.ToTensor(),
            transforms.Normalize((0.4467, 0.4398, 0.4066), (0.2603, 0.2566, 0.2713))
        ])
        return tform

    def setup(self, stage=None):
        self.data_train = STL10(