KMnP / vpt

❄️🔥 Visual Prompt Tuning [ECCV 2022] https://arxiv.org/abs/2203.12119

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Tune with scaled batch size

wangdaha1 opened this issue · comments

Hi Dr. Jia,

I noticed that in tune_fgvc/tune_vtab.py, the learning rate is scaled as 【lr = lr / 256 * cfg.DATA.BATCH_SIZE】for choosing the best learning rate while in train.py, this operation is not used (lr is kept as [5, 10, 50, etc]). I wonder the reason for doing this. Are the reported results based on the unscaled learning rate? (lr is given in your fgvc excel file)

Best,

hi, thanks for the question. we use tune*.py for all of our experiments, so we set the scaled lr in the tune_vtab.py file. So the reported results are all based on the scaled learning rate.

train.py does not set the scaling operation since it is only used as a main gate way for training. If you want to use this one directly, please remember to do the scaling manually.

Thanks for your information! I will close the question.