hi
i have a dataset that each image has a ground truth image .each ground truth image has 3 labels ,label1 is background label 2 is organ A and label 3 is organ B.
when i am using cross_entropy as loss function
criterion =cross=nn.functional.cross_entropy

this error occure
( I ran my script with CUDA_LAUNCH_BLOCKING=1 and the error log is as follows )

Screenshot from 2018-11-20 23-43-36.png823×516 50.6 KB

Could you please help me to figure out what is the problem?