code accompanying the paper RegularFace: Deep Face Recognition via Exclusive Regularization.
- Download the CASIA-webface dataset (you may request for access);
- Preprocess the dataset with code provided in https://github.com/wy1iu/sphereface/tree/master/preprocess/code;
- Train baseline methods (softmax, NormFace) with
python -m torch.distributed.launch --master_port 8889 --nproc_per_node=$GPUs pytorch/train_baseline.py - (optionally) you can train baseline methods (softmax, NormFace) with
python -m torch.distributed.launch --master_port 8889 --nproc_per_node=$GPUs pytorch/train_sample_exclusive.py.