Target 1 Is Out Of Bounds. Cross Entropy Loss