I see what you mean.
The issue is that the dataset is 80GB which would blow up Google Drive.
Is there a way to save the resized/cropped images to then pass them through the random transforms?
Here’s what I’m using:
data_transforms = {
    'train': transforms.Compose([
        transforms.Resize(256),
        transforms.CenterCrop(224),
        transforms.ToTensor(),
        transforms.Normalize([0.6000, 0.3946, 0.6041], [0.2124, 0.2335, 0.2360])
    ]),