Problem with running model

Help and Support
#1

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

image
image890×507 8.75 KB

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

#2

The problem is solved but I got another problem as follow

image
image1229×598 15.6 KB

#3

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