Solved by promach in post #4 
              
              
                I have solved all problems and it looks like I can do the predict.py without any more problems 
Please see the updated code below: 
#Modified from https://github.com/amrit-das/Custom-Model-Training-PyTorch/blob/master/predict.py
import torch
import torch.nn as nn
#from torchvision.models import resn…