Update README.md
Browse files
README.md
CHANGED
@@ -3,7 +3,7 @@ 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 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 |
|
|
|
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 |
|