Hi @NeuralFoX
Seems like you have the sequence of ascending numbers in
self.trgt=list(iter*1 for iter in range(high));
The cross_entropy_loss treats it like multi-class labels. Probably you have a buggy code in the snippet above