landskape-ai / triplet-attention

Official PyTorch Implementation for "Rotate to Attend: Convolutional Triplet Attention Module." [WACV 2021]
https://openaccess.thecvf.com/content/WACV2021/html/Misra_Rotate_to_Attend_Convolutional_Triplet_Attention_Module_WACV_2021_paper.html
MIT License
406 stars 49 forks source link

The weights in google drive is demage!!! #14

Closed shaojiestar closed 3 years ago

shaojiestar commented 3 years ago

hello my friend!!! When I download the weight in your readme.md file ,a .rar file was downloaded but I can't open it or unzip it. the error is : image

shaojiestar commented 3 years ago

so can you give me the pth file by email? 2422656558@qq.com thank you !!!

iyaja commented 3 years ago

I've faced this issue before with other models. Actually, you don't need to untar the file we provide on Google drive. You can just write something like

torch.load("RESNET50_PCAM_IMAGENET_model_best.tar")

Take a look at our scripts and commands mentioned in the README for a clear example, and let me know if you're still facing issues after this.