search

vandit15 / Class-balanced-loss-pytorch

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

typo #17

Open e-yi opened 3 years ago

e-yi commented 3 years ago

https://github.com/vandit15/Class-balanced-loss-pytorch/blob/921ccb8725b1eb0903b2c22a1a752a594fcae138/class_balanced_loss.py#L28

should be alpha: A float tensor of size [num_classes]

fcakyon commented 2 years ago

@e-yi refer to https://github.com/fcakyon/balanced-loss for improved and maintained version