Solved by Mona_Jalal in post #5 
              
              
                Thanks for the response. The error I had raises from not having this line that I added: 
model_conv = model_conv.cuda() 
from PIL import Image
scaler = transforms.Scale((224, 224))
normalize = transforms.Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225])
to_tensor = transforms.ToTensor…