[solved] Assertion `cur_target >= 0 && cur_target < n_classes' failed

2

It’s may be your linear layer’s demension problem.Try squeeze or unsqueee it.

1 Like