Assertion `cur_target >= 0 && cur_target < n_classes' failed

lizhengwei1992 / Semantic_Human_Matting

Semantic Human Matting

532 stars 146 forks source link

Open Tomhouxin opened 5 years ago

Tomhouxin commented 5 years ago

L_t = criterion(trimap_pre, trimap_gt[:,0,:,:].long()) got error

GengCauWong commented 4 years ago

L_t = criterion(trimap_pre, trimap_gt[:,0,:,:].long()) got error

I have the same problem as you. How do you solve the problem?