CUDA error: device-side assert triggered, The loss here is a tensor, and no value is returned.

[tian-003]

CUDA error: device-side assert triggered

The iseg2017 data is used, and the data format is hdr. The loss here is a tensor, and no value is returned. How to solve this problem?
run FewShot_GAN-Unet3D-master/pytorch/agents/supervised_baseline.py

self.criterion =nn.CrossEntropyLoss(class_weights)
loss = self.criterion(output_logits, labels)
loss.backward()
self.optimizer.step()