Questioin on losses.py

[ZongWei-HUST]

Hello, Thank you for sharing your great work!
I noticed that on line 36 of the losses.py file,
coefficients = torch.reciprocal(class_occurence) * torch.numel(label) / (num_of_classes * label.shape[1])
it seems that label.shape[1] should be modified to label.shape[0] here so that the solution is in line with the equation in the paper (HxW), not sure if I understand it correctly? Or is there some detail I'm not noticing?
Thanks.