Open wpd-mlb opened 2 years ago
CurricularFace/head/metrics.py / target_logit = cos_theta[torch.arange(0, embbedings.size(0)), label].view(-1, 1)
CurricularFace/head/metrics.py / target_logit = cos_theta[torch.arange(0, embbedings.size(0)), label].view(-1, 1)