pytorch-loss icon indicating copy to clipboard operation
pytorch-loss copied to clipboard

class balanced loss?

Open 2533245542 opened this issue 2 years ago • 1 comments

Ever considered implementing class balanced loss? I found your repo useful and would like to see it include more.

https://github.com/vandit15/Class-balanced-loss-pytorch

2533245542 avatar Dec 22 '21 22:12 2533245542

Sorry for being so late. I was busy with other things when this issue is created. Frankly, I do not have such a plan to add it because the implementation you posted above is good and well documented.

I tried it in sementic segmentation task of cityscapes, but observed a bit drop of miou. The model I used is deeplabv3+, and I simply replace cross entropy with this loss without other changings. Maybe I should be more careful to tune the training parameters.

CoinCheung avatar Feb 23 '22 03:02 CoinCheung