Method setup

datamodule.py:214–224

(self, stage=None)

        return tform

    def setup(self, stage=None):
        self.data_train = STL10(
            root = self.data_dir,
            split = 'unlabeled', 
            transform = self.transforms(not self.augment)
        )
        self.data_val = STL10(
            root = self.data_dir,
            split = 'train', 
            transform = self.transforms(True)
        )

class STL10LinearProbeDataModule(_BaseDataModule):
    num_classes = 10