Solved by KFrank in post #2 
              
              
                Hello Over! 

Well, it depends on what you do with the labels. 
In the common case that you use CrossEntropyLoss as your loss 
function, per its documentation, your labels (the target) must be 
integer class labels that run from [0, nClass - 1], inclusive.  That 
is, the labels start from 0. 
Best. 
…