search

vandit15 / Class-balanced-loss-pytorch

Pytorch implementation of the paper "Class-Balanced Loss Based on Effective Number of Samples"
MIT License
781 stars 120 forks source link

Typo in #line87 #6

Open Anurag14 opened 4 years ago

Anurag14 commented 4 years ago

'weight' keyword for sigmoid loss spelled as 'weights'