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

2stage training #1

Closed sssssshf closed 2 years ago

sssssshf commented 2 years ago

Do you need to freeze the weight of feature network for the second stage training of classification layer???

ShadeAlsha commented 2 years ago

The goal of the second stage is classifier learning and we do freeze the feature network weights during it.