File size: 721 Bytes
3d6299a
 
c1a4fc0
3d6299a
 
c1a4fc0
3d6299a
c1a4fc0
3d6299a
 
 
c1a4fc0
 
 
3d6299a
c1a4fc0
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
---
library_name: transformers
license: apache-2.0
---

# Model Card for vit-small-patch16-224-single-channel

A Vision Transformer (ViT) image model with a single channel.

## Model Details

This model is a variant of the `vit_small_patch16_224.augreg_in21k_ft_in1k` model
from the [timm](https://huggingface.co/docs/timm/index) library that contains
a single channel instead of three.

The `timm` model was converted to a `transformers` model using a modified version of the 
[convert_vit_timm_to_pytorch.py](https://github.com/huggingface/transformers/blob/f9c7e6021e9a9a9fd3fc8bb291da9451066aeb8d/src/transformers/models/vit/convert_vit_timm_to_pytorch.py)
script which is included in this repository as `convert.py`.