kentaroy47 / vision-transformers-cifar10

Let's train vision transformers (ViT) for cifar 10!
MIT License
517 stars 106 forks source link

Is it from scratch? #33

Open free001style opened 2 weeks ago

free001style commented 2 weeks ago

Hi! Do I understand correctly that you were using pretrained models? If not, I don't see it in your code.

kentaroy47 commented 2 weeks ago

Default settings use scratch models.

2024年9月12日(木) 5:05 Ilya Drobyshevskiy @.***>:

Hi! Do I understand correctly that you were using pretrained models? If not, I don't see it in your code.

— Reply to this email directly, view it on GitHub https://github.com/kentaroy47/vision-transformers-cifar10/issues/33, or unsubscribe https://github.com/notifications/unsubscribe-auth/AJOG4JPYPMCJTKPUYDN6KALZWCPCLAVCNFSM6AAAAABOBWOVL6VHI2DSMVQWIX3LMV43ASLTON2WKOZSGUZDANRWGE3TSOI . You are receiving this because you are subscribed to this thread.Message ID: @.***>

kentaroy47 commented 2 weeks ago

python train_cifar10.py --net vit_tim

This uses pertained models.

2024年9月12日(木) 8:18 @. @.>:

Default settings use scratch models.

2024年9月12日(木) 5:05 Ilya Drobyshevskiy @.***>:

Hi! Do I understand correctly that you were using pretrained models? If not, I don't see it in your code.

— Reply to this email directly, view it on GitHub https://github.com/kentaroy47/vision-transformers-cifar10/issues/33, or unsubscribe https://github.com/notifications/unsubscribe-auth/AJOG4JPYPMCJTKPUYDN6KALZWCPCLAVCNFSM6AAAAABOBWOVL6VHI2DSMVQWIX3LMV43ASLTON2WKOZSGUZDANRWGE3TSOI . You are receiving this because you are subscribed to this thread.Message ID: @.***>