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

Update class_balanced_loss.py

Open krittaprot opened this issue 1 year ago • 1 comments

The typo of weight parameter causes an error. It should be "weight" and not "weights".

krittaprot avatar Feb 18 '24 05:02 krittaprot

@krittaprot feel free to use this improved/fixed/battle-tested version, includes this fix: https://github.com/fcakyon/balanced-loss#improvements

fcakyon avatar Dec 17 '24 11:12 fcakyon