gent commited on
Commit
f38ae12
1 Parent(s): ed01e82

Update README.md with SiT model information

Browse files
Files changed (1) hide show
  1. README.md +40 -0
README.md CHANGED
@@ -1,3 +1,43 @@
1
  ---
2
  license: apache-2.0
 
 
 
 
 
 
3
  ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  ---
2
  license: apache-2.0
3
+ tags:
4
+ - self-supervised learning
5
+ - vision
6
+ - SiT
7
+ inference: false
8
+
9
  ---
10
+
11
+ # Model description
12
+
13
+ SiT is a self-supervised learning model that combines masked image modeling and contrastive learning. The model is trained on ImageNet-1K.
14
+
15
+
16
+ # Model Sources
17
+
18
+ - https://github.com/Sara-Ahmed/SiT
19
+ - https://arxiv.org/abs/2104.03602
20
+
21
+
22
+ # Model Card Authors
23
+ Sara Atito, Muhammad Awais, Josef Kittler
24
+
25
+ # How to use
26
+
27
+ ```python
28
+ from modeling_sit import ViTSiTForPreTraining
29
+ # reload
30
+ model = ViTSiTForPreTraining.from_pretrained("erow/SiT")
31
+ ```
32
+
33
+ # BibTeX entry and citation info
34
+ ```
35
+ @inproceedings{atito2023sit,
36
+ title={SiT is all you need},
37
+ author={Atito, Sara and Awais, Muhammed and Nandam, Srinivasa and Kittler, Josef},
38
+ booktitle={2023 IEEE International Conference on Image Processing (ICIP)},
39
+ pages={2125--2129},
40
+ year={2023},
41
+ organization={IEEE}
42
+ }
43
+ ```