Thanks, it’s working now.
I had one more question though. It seems that my transforms library doesn’t have RandomVerticalFlip and a few other transforms. However I had built PyTorch from source just a few weeks back. Do I need to build from source again?