PyTorch Forums
Unsolved: NLLLoss throws Assertion `t >= 0 && t < n_classes` failed (when batchsize>1)
Botho_Quinn
(Botho Quinn)
December 1, 2021, 10:37am
3
Also here is the cuda error after adding ‘CUDA_LAUNCH_BLOCKING’ = 1
cudaassert
1632×207 21.4 KB
show post in topic