Runtimeerror: Mat1 Dim 1 Must Match Mat2 Dim 0 With Addition Of Data Augmentation

I am getting this error in the training loop. The model was training fine before I added an augmentation function in the datasets.py file. I can't understand what it is. My model i

Solution 1:

Update -: I fixed the error by resizing the image in datasets.py code.