deepinsight / deepinsight/insightface
CurricularFace loss
- Dominant language
- Python
- Stars
- 29.7k
- Forks
- 6.1k
- PR merge metrics
- No merged PRs in 30d
Description
could author add CurricularFace loss to your insightface?
CurricularFace loss: https://github.com/HuangYG123/CurricularFace/blob/master/head/metrics.py
it was occur error to mask = cos_theta > cos_theta_m with multi gpu
RuntimeError: The size of tensor a (6387) must match the size of tensor b (110) at non-singleton dimension 1
Contributor guide
No contributing guide indexed for this repository
Research direction
Start by reading the referenced CurricularFace implementation in head/metrics.py and reproduce the reported multi-GPU error involving mask = cos_theta > cos_theta_m. Trace how the loss is integrated into InsightFace and verify that the added loss works without the tensor-size mismatch in multi-GPU training.
Written by the indexing model from the issue text.
Assessment
- Tech stack
- python, pytorch
- Domain
- computer-vision, machine-learning
- Issue type
- Feature
- Difficulty
- 4/5
- Estimated time
- 3-5 days
- Activity status
- Stale
- Clarity
- Mostly clear
- Newbie friendliness
- 35/100