Solved by ptrblck in post #2 
              
              
                The training loss looks alright. I used the provided shapes to check the input and target and think y = (y.long()).view(-1) should be unnecessary. Could you verify it? 
The test loss might be alright. I’m not too familiar with your use case, but note that you are not applying the normalization. Yo…