Giters

lucidrains / vit-pytorch

Implementation of Vision Transformer, a simple way to achieve SOTA in vision classification with only a single transformer encoder, in Pytorch

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

where is the train.py? tks

henbucuoshanghai opened this issue · comments

commented

Have you solved this problem? I have the same problem

commented

There is neither a training script nor are there pre-trained models. This is a repository that provides implementations of different ViT architectures. If you need an example how to train a model from scratch, check out the provided notebook. Basically, can also use any other tutorial that provides information on training classifiers in PyTorch (e.g., this one) and just swap the classifier (however, keep in mind that depending on your data, additionally modifications might be necessary).