ckpt file not found for image classification

dingmyu / davit

[ECCV 2022]Code for paper "DaViT: Dual Attention Vision Transformer"

MIT License

329 stars 33 forks source link

Open ghost opened 2 years ago

ghost commented 2 years ago

Hi, I download checkpoints from 'Image Classification on ImageNet-1K', which are data in fact after decompressed. So is there something wrong here?

dingmyu commented 2 years ago

Hi,

The file extension is ‘pth.tar’. You can directly load the model by the 'torch.load' function. Feel free to post here if there are other questions.

Regards, Mingyu