ofsoundof / LocalViT

MIT License
113 stars 14 forks source link

LocalVit-Swin #4

Open caihaunqai opened 2 years ago

caihaunqai commented 2 years ago

Hello, Thanks for your excellent job. I wonder is it better for swintransformer to add Local module paper proposed? I can not find this conclution from the paper or github. Thank you

ofsoundof commented 2 years ago

Thanks for your interest in our paper. The results of applying the proposed locality module to Swin transformer are given in the readme of this repository. The code for that is given at

https://github.com/ofsoundof/LocalViT/blob/main/models/localvit_swin.py#L242

We will update the arxiv version of the paper.