typo

[e-yi]

Class-balanced-loss-pytorch/class_balanced_loss.py

Line 28 in 921ccb8

alpha: A float tensor of size [batch_size]

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

[fcakyon]

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