Reproduce face recognition model training

[and86802]

Thank you for the awesome work!

I'm now reproducing the result using the synthetic dataset you provided.
May I know how you train the FR model?
I'm using the pipeline the same as AdaFace, but seems it need some parameter tuning cause I got loss=nan when initial lr=0.1...

Thank you very much!