Solved by ptrblck in post #2 
              
              
                You are passing the transforms.Normalize object to the network instead of applying it on the input data. 
Pass the input to transforms.Normalize and pass the return value to the model.