Skip to content

Conversation

@rwightman
Copy link
Collaborator

No description provided.

@amaarora
Copy link
Contributor

amaarora commented May 18, 2021

@rwightman
Commenting here again as this is a better place for it.

You should be able to find visformer_small weights here - https://www.kaggle.com/aroraaman/visformer-small-chkpt

Val accuracy - 82.1%, paper reports - 82.28%. Val img size - 224x224.

Please note that I didn't train the model myself but copied the weights from here and edited the keys so that the pretrained weights can be loaded to TIMM.

Hope this helps :)

@LvJC
Copy link

LvJC commented May 19, 2021

Where's the RedNet pretrained weights?
I load rednet50ts but get a warning:

No pretrained weights exist for this model. Using random initialization.

@rwightman
Copy link
Collaborator Author

@LvJC clearly they don't exist, this is a development branch

@rwightman rwightman merged commit 07d952c into master May 25, 2021
@rwightman rwightman deleted the levit_visformer_rednet branch November 19, 2021 16:47
guoriyue pushed a commit to guoriyue/pytorch-image-models that referenced this pull request May 24, 2024
LeVit, Visformer, RedNet/Involution models and layers
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

4 participants