ShadeAlsha / LTR-weight-balancing

CVPR 2022 - official implementation for "Long-Tailed Recognition via Weight Balancing" https://arxiv.org/abs/2203.14197
MIT License
118 stars 10 forks source link

Some implementation confusion about ImageNet and iNaturalist #13

Open cool-xuan opened 1 year ago

cool-xuan commented 1 year ago

I really appreciate your work and am grateful for your open-sourcing of the code.
I was able to successfully reproduce the experimental results on CIFAR100-LT using your code, but I was wondering if you could provide me with some additional information on the hyperparameter settings for the larger datasets, ImageNet-LT and iNaturalist.

Specifically, I would greatly appreciate it if you could share the following:

For Stage 1: the initial learning rate, epoch number, weight decay settings, and any data augmentation techniques used (such as color jittering for ImageNet).

For Stage 2: the initial learning rate, epoch number, weight decay settings, and the hyperparameters in CBLoss (loss type, beta, and gamma).

Thank you once again for your fantastic work and for your generosity in sharing your code with the community.