SHI-Labs / Neighborhood-Attention-Transformer

Neighborhood Attention Transformer, arxiv 2022 / CVPR 2023. Dilated Neighborhood Attention Transformer, arxiv 2022
MIT License
1.05k stars 86 forks source link

unable to get the pretrain model for classification #18

Closed zhou-rui1 closed 2 years ago

zhou-rui1 commented 2 years ago

Hi, it seems that the link for the NAT-base classification model is garbled code, could you fix it or send it to me with 563937790@qq.com?

Thanks!

alihassanijr commented 2 years ago

Hi, thank you for your interest. I'm not sure what the issue is, the links work fine for me. What exactly is the error you observe when you try to use them?

zhou-rui1 commented 2 years ago

well, when I click the link of eg.[84.3%], it looks like this:(http://ix.cs.uoregon.edu/~alih/nat/checkpoints/CLS/nat_base.pth)

image
stevenwalton commented 2 years ago

You need to download the file. You can refer to the torch documents for how to utilize this file. https://pytorch.org/tutorials/beginner/saving_loading_models.html