Question about the loss

Gavinwxy / DGCL

[CVPR 2023] Hunting Sparsity: Density-Guided Contrastive Learning for Semi-Supervised Semantic Segmentation

Apache License 2.0

23 stars 1 forks source link

Question about the loss #13

Open Hugo-cell111 opened 9 months ago

Hugo-cell111 commented 9 months ago

Hi! When I reproduce your code, I find there is a bug in the sup loss calculation: RuntimeError: CUDA error: device-side assert triggered /opt/conda/conda-bld/pytorch_1659484657607/work/aten/src/ATen/native/cuda/NLLLoss2d.cu:103: nll_loss2d_forward_kernel: block: [2,0,0], thread: [640,0,0] Assertion t >= 0 && t < n_classes failed. Could you please provide some suggestions?