소스 검색

add YOLOX-Plus

yjh0410 2 년 전
부모
커밋
bc8f08692b
1개의 변경된 파일0개의 추가작업 그리고 1개의 파일을 삭제
  1. 0 1
      models/detectors/yolox_plus/loss.py

+ 0 - 1
models/detectors/yolox_plus/loss.py

@@ -136,7 +136,6 @@ class ClassificationLoss(nn.Module):
             gt_score: (torch.Tensor): [N, C]
         """
         gt_label = gt_label.long()
-        gt_score = gt_score[:]
         gt_score = gt_score[torch.arange(gt_label.shape[0]), gt_label]
 
         pred_sigmoid = pred_cls.sigmoid()