Thank you so much for this! I am completely new to PyTorch so I knew I was doing something silly. Just did not know what it was. This worked out just fine!
I was wondering if you could tell me why it showed my accuracy going up with MSE loss when I had one-hot labels?
Thank you again.