Right now I’m currently using this for the transformations of my images before feeding them into my CNN for training:
   self.transform = transforms.Compose([
        transforms.ToPILImage(),
        transforms.RandomCrop(60),
        transforms.RandomHorizontalFlip(),
        transforms.ToTensor()
    ])

which is located in my IcebergDataset class which is a subclass of torch.utils.data.Dataset.
During testing, I am still using this transform code which to me seems a bit weird as I’m augmenting my test set images. Am I using this functionality the right way?