mAUCMLoss for multi-label classification when bs=1 loss return 0

Optimization-AI / LibAUC

LibAUC: A Deep Learning Library for X-Risk Optimization

https://libauc.org/

MIT License

273 stars 37 forks source link

mAUCMLoss for multi-label classification when bs=1 loss return 0 #37

Closed SNEAndy closed 7 months ago

SNEAndy commented 10 months ago

Hi,I'm using the mAUCMLoss to finish my multi-label classification. I realized that if the batch size of my dataloader is 1,the mAUCMLoss will always be 0,otherwise,it will not. So the batch size must greater than 1 to fit this loss function ?

optmai commented 7 months ago

Do you still have this issue?