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

Published 20 hours ago • verified publisher icon vandit15

typo

Open e-yi opened this issue 3 years ago • 1 comments

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]

e-yi avatar Jul 26 '21 09:07 e-yi

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

fcakyon avatar Jul 22 '22 13:07 fcakyon