from PIL import Image
from IPython.display import display
import matplotlib.image as mpimg
img = mpimg.imread('train/train2/plastic472.jpg')
imgplot = plt.imshow(img)
image= Image.open('train/train2/plastic472.jpg')
plt.show()
im= transform(image)
net = ConvNet()
net.load_state_dict(torch.load(PATH))
outputs = net(im)

I tried to test my NN, but I got this error, no such errors were while training