class-balanced-loss
class-balanced-loss copied to clipboard
My implementation in Pytorch doesn't work
Dear authors, thanks for your greate effort to make your code open-source. I re-implement you CB-focal Loss in Pytorch(Both in your tf-version and my own version), but can't reach the performance reported in your paper.
This is my code. Could you please have a check whether there is something wrong with my code? output: [batch_size, num_classes] label: [batch_size] catList: [num_classes] a list of sampler numbers for each class
Hi,
I think your code is fine. I looked it carefully, and I didn't find any problem. Did you reach the performance reported in the paper? Thanks a lot
Hi,
I think your code is fine. I looked it carefully, and I didn't find any problem. Did you reach the performance reported in the paper? Thanks a lot
I finished my experiments on long-tail CIFAR10 which author released in TF-Record mode but didn't achieved the acc reported in the paper.
Same issue here as well. (I also used Pytorch for re-implementation) I hope the authors provide more details and trained parameters for long-tailed CIFAR.
@hbsz123 Please let us know if you have any progress.
Quick question : catList: [num_classes] a list of sampler numbers for each class
sampler number for each class here is calculated for each batch or entire dataset?