Solved by Chirath_Pansilu in post #2 
              
              
                F.cross_entropy needs labels to be long tensors. You have passed in a float tensor. So in the validation_step function add the code 
labels = labels.long() to convert it into a long tensor