Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Add new backbones trained with registers #282

Merged
merged 2 commits into from
Oct 27, 2023
Merged

Add new backbones trained with registers #282

merged 2 commits into from
Oct 27, 2023

Conversation

patricklabatut
Copy link
Contributor

Add new backbones (and matching linear classification heads) trained with 4 registers following Vision Transformers Need Registers.

@facebook-github-bot facebook-github-bot added the CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. label Oct 27, 2023
@patricklabatut patricklabatut merged commit 9c7e324 into main Oct 27, 2023
2 checks passed
@patricklabatut patricklabatut deleted the vitneedreg branch October 27, 2023 13:15
@patricklabatut patricklabatut linked an issue Oct 27, 2023 that may be closed by this pull request
@mshooter
Copy link

Hi I am getting this error when try to load it from the hub..
RuntimeError: Cannot find callable dinov2_vitb14_reg in hubconf

@patricklabatut
Copy link
Contributor Author

patricklabatut commented Oct 28, 2023

RuntimeError: Cannot find callable dinov2_vitb14_reg in hubconf

@mshooter If you previously used the provided models via PyTorch Hub before this new release, then an older version of the repo may simply be cached by PyTorch (under ~/.cache/torch/hub). Try passing force_reload=True to torch.hub.load() to bypass the cached version (alternatively remove the cached repo under ~/.cache/torch/hub).

@mshooter
Copy link

@patricklabatut, the force_reload=True fixed it for me.
Thank you!

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
Projects
None yet
Development

Successfully merging this pull request may close these issues.

DINOv2 ViT Registers
4 participants