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
this error occure
( I ran my script with
CUDA_LAUNCH_BLOCKING=1 and the error log is as follows )
Could you please help me to figure out what is the problem?