This repository is aimed to implement a minimal ArcFace (https://arxiv.org/pdf/1801.07698.pdf) based on PyTorch framework. The repository content may alter from time to time.
single_GPU
- Single GPU training
ddp
- Distributed Data Parallel (DDP) training
If you are facing any problems with repository, please open an issue for a detailed discussion.