Keras
TF-Keras
sayakpaul HF staff commited on
Commit
aee4c1c
1 Parent(s): 890132b

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -2
README.md CHANGED
@@ -2,9 +2,10 @@
2
  library_name: keras
3
  ---
4
 
5
- This model is a TensorFlow port of DINO ViT B-16 [1]. The backbone of this model was pre-trained using the DINO pretext task. After that its head layer was trained
6
  by keeping the backbone frozen. ImageNet-1k dataset was used for training purposes. You can refer to [this notebook](https://github.com/sayakpaul/probing-vits/blob/main/notebooks/load-dino-weights-vitb16.ipynb) to know how the porting was done.
7
 
8
  ## References
9
 
10
- [1] Emerging Properties in Self-Supervised Vision Transformers: https://arxiv.org/abs/2104.14294
 
 
2
  library_name: keras
3
  ---
4
 
5
+ This model is a TensorFlow port of DINO [1] ViT B-16 [2]. The backbone of this model was pre-trained using the DINO pretext task. After that its head layer was trained
6
  by keeping the backbone frozen. ImageNet-1k dataset was used for training purposes. You can refer to [this notebook](https://github.com/sayakpaul/probing-vits/blob/main/notebooks/load-dino-weights-vitb16.ipynb) to know how the porting was done.
7
 
8
  ## References
9
 
10
+ [1] Emerging Properties in Self-Supervised Vision Transformers: https://arxiv.org/abs/2104.14294
11
+ [2] An Image is Worth 16x16 Words: Transformers for Image Recognition at Scale: https://arxiv.org/abs/2010.11929