Thanks a ton!
So to make sure I understand correctly, I shouldn’t take the argmax of the output and should reshape the argmax of the target to (N,) instead of (N,1)?
If so, how would I properly reshape target? I tried .view(n,) and it doesn’t appear PyTorch supports that.