-
- Downloads
Add new backbones trained with registers (#282)
Add new backbones (and matching linear classification heads) trained with 4 registers following [Vision Transformers Need Registers](https://arxiv.org/abs/2309.16588).
Showing
- MODEL_CARD.md 79 additions, 8 deletionsMODEL_CARD.md
- README.md 125 additions, 0 deletionsREADME.md
- dinov2/configs/eval/vitb14_reg4_pretrain.yaml 9 additions, 0 deletionsdinov2/configs/eval/vitb14_reg4_pretrain.yaml
- dinov2/configs/eval/vitg14_reg4_pretrain.yaml 10 additions, 0 deletionsdinov2/configs/eval/vitg14_reg4_pretrain.yaml
- dinov2/configs/eval/vitl14_reg4_pretrain.yaml 9 additions, 0 deletionsdinov2/configs/eval/vitl14_reg4_pretrain.yaml
- dinov2/configs/eval/vits14_reg4_pretrain.yaml 9 additions, 0 deletionsdinov2/configs/eval/vits14_reg4_pretrain.yaml
- dinov2/configs/ssl_default_config.yaml 3 additions, 0 deletionsdinov2/configs/ssl_default_config.yaml
- dinov2/hub/backbones.py 76 additions, 4 deletionsdinov2/hub/backbones.py
- dinov2/hub/classifiers.py 135 additions, 14 deletionsdinov2/hub/classifiers.py
- dinov2/hub/utils.py 3 additions, 2 deletionsdinov2/hub/utils.py
- dinov2/models/__init__.py 3 additions, 0 deletionsdinov2/models/__init__.py
- dinov2/models/vision_transformer.py 39 additions, 7 deletionsdinov2/models/vision_transformer.py
- hubconf.py 2 additions, 0 deletionshubconf.py
Loading
Please register or sign in to comment