https://github.com/damo-cv/MogFace/blob/5e064d6669b50554900ba79c440b99931052f50e/modelling/pred_modules/pred_net.py#LL104C63-L104C63
