search

yan-hao-tian / VW

iclr2024 poster Varying Window Attention
MIT License
114 stars 19 forks source link

imagenet train code #9

Closed endaoguansanlu closed 2 months ago

endaoguansanlu commented 1 year ago

Hi! Thanks for your excellent work, do you have a plan to release the imagenet pre-train code? ^ _ ^

yan-hao-tian commented 1 year ago

https://drive.google.com/drive/folders/1b7bwrInTW4VLEm27YawHOAMSMikga2Ia?usp=sharing Hi! BTW, the pretrained MITs are published by SegFormer (github.com/NVlabs/SegFormer). The pre-training code may be found there.