My output is in the right format, (N, C) = (64, 9). The issue is that since I have 9 classes, the loss function by default take/expects labels 0-8. However, since I only get rid of number “6”, the labels left is “1,2,3,4,5,7,8,9”. The “9” exceed the loss expectation.