vision-transformer-pytorch icon indicating copy to clipboard operation
vision-transformer-pytorch copied to clipboard

parameter setting to train on ImageNet from scratch

Open zhiyuanli1992 opened this issue 4 years ago • 4 comments

Thank you for your great work. Would you please post your training scripts to train the model on ImageNet. For example the learning rate, weight decay, training_step and etc? Thank you very much!

zhiyuanli1992 avatar Mar 20 '21 21:03 zhiyuanli1992

+1

DevilYangS avatar Mar 30 '21 09:03 DevilYangS

Did you actually try to train your model from scratch, or did you always use the pre-trained weights?

In particular, in the README, you write

Similar results as in original implementation are achieved.

Do you mean, when training from scratch, or when importing pre-trained weights and fine-tuning?

cjsg avatar Mar 30 '21 14:03 cjsg

The cifar results are fine-tuned from the pretrained weights, not training from scratch

Hhhhhhao avatar Mar 31 '21 02:03 Hhhhhhao

ar results are fine-tuned from the pretrained weights, not training from scratch

Hi, can I have the training script to train on ImageNet from scratch? Thanks!

zhiyuanli1992 avatar Mar 31 '21 16:03 zhiyuanli1992