an error

[caogule]

Hi when i run your code ，i find a problem，could you give me a hand? how to solve the problem ?

/pytorch/torch/lib/THCUNN/ClassNLLCriterion.cu:101: void cunn_ClassNLLCriterion_updateOutput_kernel(Dtype *, Dtype *, Dtype *, long *, Dtype *, int, int, int, int, long) [with Dtype = float, Acctype = float]: block: [0,0,0], thread: [15,0,0] Assertion t >= 0 && t < n_classes failed. THCudaCheck FAIL file=/pytorch/torch/lib/THC/generic/THCStorage.c line=36 error=59 : device-side assert triggered Traceback (most recent call last): File "main_recog.py", line 213, in train(dataloader, netE, criterionCE, optimizerE, epoch) File "main_recog.py", line 187, in train losses.update(loss.data[0], input.size(0)) RuntimeError: cuda runtime error (59) : device-side assert triggered at /pytorch/torch/lib/THC/generic/THCStorage.c:36