fix communication and gamma issues in contrastive loss - Githubissues

Optimization-AI / LibAUC

LibAUC: A Deep Learning Library for X-Risk Optimization

https://libauc.org/

MIT License

273 stars 37 forks source link

fix communication and gamma issues in contrastive loss #53

Closed xywei00 closed 4 months ago

xywei00 commented 4 months ago

fix incorrect communication with all_gather in GCLoss_v1
set gamma to original value when u is not 0