Image Classification
MLX
Safetensors
data2vec-vision
vision
SauravMaheshkar commited on
Commit
b1d189a
·
verified ·
1 Parent(s): 46f1a7d

docs: add usage instructions

Browse files
Files changed (1) hide show
  1. README.md +11 -1
README.md CHANGED
@@ -13,4 +13,14 @@ datasets:
13
 
14
  ![model image](https://raw.githubusercontent.com/patrickvonplaten/scientific_images/master/data2vec.png)
15
 
16
- BEiT model pre-trained in a self-supervised fashion and fine-tuned on ImageNet-1k (1,2 million images, 1000 classes) at resolution 224x224. It was introduced in the paper [data2vec: A General Framework for Self-supervised Learning in Speech, Vision and Language](https://arxiv.org/abs/2202.03555) by Alexei Baevski, Wei-Ning Hsu, Qiantong Xu, Arun Babu, Jiatao Gu, Michael Auli and first released in [this repository](https://github.com/facebookresearch/data2vec_vision/tree/main/beit).
 
 
 
 
 
 
 
 
 
 
 
13
 
14
  ![model image](https://raw.githubusercontent.com/patrickvonplaten/scientific_images/master/data2vec.png)
15
 
16
+ BEiT model pre-trained in a self-supervised fashion and fine-tuned on ImageNet-1k (1,2 million images, 1000 classes) at resolution 224x224. It was introduced in the paper [data2vec: A General Framework for Self-supervised Learning in Speech, Vision and Language](https://arxiv.org/abs/2202.03555) by Alexei Baevski, Wei-Ning Hsu, Qiantong Xu, Arun Babu, Jiatao Gu, Michael Auli and first released in [this repository](https://github.com/facebookresearch/data2vec_vision/tree/main/beit).
17
+
18
+ ## Usage
19
+
20
+ ```python
21
+ from mlx_ssl.models import Data2VecVisionForImageClassification
22
+
23
+ model = Data2VecVisionForImageClassification.from_pretrained(
24
+ "mlx-community/data2vec-vision-large-ft1k"
25
+ )
26
+ ```