search

cvqluu / Angular-Penalty-Softmax-Losses-Pytorch

Angular penalty loss functions in Pytorch (ArcFace, SphereFace, Additive Margin, CosFace)
MIT License
482 stars 91 forks source link

sum instead of mean? #16

Open AngledLuffa opened 1 year ago

AngledLuffa commented 1 year ago

Would you be up for adding a reduction=... option which allows for sum vs mean?