def METHOD_NAME(self, stage: str):
    transform = transforms.Compose(
        [transforms.Pad(4), transforms.RandomHorizontalFlip(), transforms.RandomCrop(32), transforms.ToTensor()]
    )
    self.train_dataset = CIFAR10(root=self.train_path, train=True, transform=transform, download=False)
    self.val_dataset = CIFAR10(root=self.vld_path, train=True, transform=transform, download=False)