Loss function APIs for face recognition


Could you guys recommend some loss functions (better have pytorch API provided) that may be useful to face recognition?

I’ve gone through all the loss function APIs but not sure which one is robust for face recognition.
I know TripletMarginLoss may be good, but I cannot find examples on implementation of that. So can you recommend some other?

Thanks a lot!