Unsolved: NLLLoss throws Assertion `t >= 0 && t < n_classes` failed (when batchsize>1)

Also here is the cuda error after adding ‘CUDA_LAUNCH_BLOCKING’ = 1