Thanks for your advice, I changed it, but didn’t help with the overfitting problem of my network. What do you mean by swapping the training set for the validation set?
I also tried  transforms.RandomRotation(degrees=5)  and  transforms.RandomHorizontalFlip(p=0.5) on my train images. But after 4 epochs, the val loss started rising again and the resulting test accuracy was poor.