search

berniwal / swin-transformer-pytorch

Implementation of the Swin Transformer in PyTorch.
https://arxiv.org/pdf/2103.14030.pdf
MIT License
782 stars 125 forks source link

apply to other dataset #20

Open jieweilai opened 2 years ago

jieweilai commented 2 years ago

hello,thanks for the work you had done very much and i have a question that how can i apply this code to train a vit model on other dataset,how can i to adjust those parameters?