CoolFace
Modelpublic

HAMMALE/vit-tiny-classifier-rvlcdip

sourceHugging Faceapache-2.0updated 1y agoView on Hugging Face
0likes152downloads
config.json62 linesDownload Raw Back to root
1{2  "architectures": [3    "ViTForImageClassification"4  ],5  "attention_probs_dropout_prob": 0.0,6  "encoder_stride": 16,7  "hidden_act": "gelu",8  "hidden_dropout_prob": 0.0,9  "hidden_size": 192,10  "id2label": {11    "0": "letter",12    "1": "form",13    "2": "email",14    "3": "handwritten",15    "4": "advertisement",16    "5": "scientific_report",17    "6": "scientific_publication",18    "7": "specification",19    "8": "file_folder",20    "9": "news_article",21    "10": "budget",22    "11": "invoice",23    "12": "presentation",24    "13": "questionnaire",25    "14": "resume",26    "15": "memo"27  },28  "image_size": 224,29  "initializer_range": 0.02,30  "intermediate_size": 768,31  "label2id": {32    "advertisement": 4,33    "budget": 10,34    "email": 2,35    "file_folder": 8,36    "form": 1,37    "handwritten": 3,38    "invoice": 11,39    "letter": 0,40    "memo": 15,41    "news_article": 9,42    "presentation": 12,43    "questionnaire": 13,44    "resume": 14,45    "scientific_publication": 6,46    "scientific_report": 5,47    "specification": 748  },49  "layer_norm_eps": 1e-12,50  "model_type": "vit",51  "num_attention_heads": 3,52  "num_channels": 3,53  "num_hidden_layers": 12,54  "patch_size": 16,55  "pooler_act": "tanh",56  "pooler_output_size": 192,57  "problem_type": "single_label_classification",58  "qkv_bias": true,59  "torch_dtype": "float32",60  "transformers_version": "4.52.4"61}62