huawei-noah / Efficient-AI-Backbones

Efficient AI Backbones including GhostNet, TNT and MLP, developed by Huawei Noah's Ark Lab.
4.07k stars 708 forks source link

Finetuning fails on ViG #176

Closed tdzdog closed 1 year ago

tdzdog commented 1 year ago

I tried to finetune on ViG but the accuracy keeps decreasing. I used the ViG tiny and the provided vig_ti_74.5.pth as the pretrained weights. I finetuned for 30 epochs and used first 5 epochs for warmup. The accuracy decreases even in the warmup stage (74.5%->73.9%). Then the accuracy even decreases to 62.5%. I used similar learning rate 2e-3 and cosine scheduler (same as pretraining). I am wondering what is going wrong?

iamhankai commented 1 year ago

During finetuing, the learning rate is usually set smaller (e.g., x0.1) than during pretraining.