Problem with running model

Hello,
I got the following problem while running the modelling for my dataset.

My notebook: https://jovian.ai/zyc191/deep-learning-projekt-jovian

The problem is solved but I got another problem as follow

It works now by resize the images to (128,128) in torchvision transform