sean1 commited on
Commit
fa20550
·
1 Parent(s): 607c494

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -11
README.md CHANGED
@@ -16,16 +16,6 @@ widget:
16
  - src: https://huggingface.co/datasets/mishig/sample_images/resolve/main/palace.jpg
17
  example_title: Palace
18
  ---
19
- # Vision Transformer (base-sized model)
20
 
21
- Vision Transformer (ViT) model pre-trained on ImageNet-21k (14 million images, 21,843 classes) at resolution 224x224, and fine-tuned on ImageNet 2012 (1 million images, 1,000 classes) at resolution 224x224. It was introduced in the paper [An Image is Worth 16x16 Words: Transformers for Image Recognition at Scale](https://arxiv.org/abs/2010.11929) by Dosovitskiy et al. and first released in [this repository](https://github.com/google-research/vision_transformer). However, the weights were converted from the [timm repository](https://github.com/rwightman/pytorch-image-models) by Ross Wightman, who already converted the weights from JAX to PyTorch. Credits go to him.
22
 
23
- Disclaimer: The team releasing ViT did not write a model card for this model so this model card has been written by the Hugging Face team.
24
-
25
- ## Model description
26
-
27
- The Vision Transformer (ViT) is a transformer encoder model (BERT-like) pretrained on a large collection of images in a supervised fashion, namely ImageNet-21k, at a resolution of 224x224 pixels. Next, the model was fine-tuned on ImageNet (also referred to as ILSVRC2012), a dataset comprising 1 million images and 1,000 classes, also at resolution 224x224.
28
-
29
- Images are presented to the model as a sequence of fixed-size patches (resolution 16x16), which are linearly embedded. One also adds a [CLS] token to the beginning of a sequence to use it for classification tasks. One also adds absolute position embeddings before feeding the sequence to the layers of the Transformer encoder.
30
-
31
- By pre-training the model, it learns an inner representation of images that can then be used to extract features useful for downstream tasks: if you have a dataset of labeled images for instance, you can train a standard classifier by placing a linear layer on top of the pre-trained encoder. One typically places a linear layer on top of the [CLS] token, as the last hidden state of this token can be seen as a representation of an entire image.
 
16
  - src: https://huggingface.co/datasets/mishig/sample_images/resolve/main/palace.jpg
17
  example_title: Palace
18
  ---
19
+ # Efficientnetv2 (base-sized model)
20
 
 
21