The rotate transform seems to generate images with random values in some pixels and sometimes even returns Nans in the image, thereby leading to issues in training.
An example of using the rotate transform (top row are the original Cifar-10 images and bottom row shows rotated images):
The rotate transform seems to generate images with random values in some pixels and sometimes even returns Nans in the image, thereby leading to issues in training.
An example of using the rotate transform (top row are the original Cifar-10 images and bottom row shows rotated images):