{ "_name_or_path": "google/vit-base-patch16-224-in21k", "architectures": [ "ViTForImageClassification" ], "attention_probs_dropout_prob": 0.0, "encoder_stride": 16, "hidden_act": "gelu", "hidden_dropout_prob": 0.0, "hidden_size": 768, "id2label": { "0": "BERET", "1": "FEDORA", "2": "SOMBERO", "3": "HARD HAT", "4": "FEZ", "5": "ZUCCHETTO", "6": "TOP HAT", "7": "DEERSTALKER", "8": "ASCOT CAP", "9": "PORK PIE", "10": "MILITARY HELMET", "11": "BICORNE", "12": "FOOTBALL HELMET", "13": "MOTARBOARD", "14": "BOATER", "15": "PITH HELMET", "16": "SOUTHWESTER", "17": "BOWLER", "18": "GARRISON CAP", "19": "BASEBALL CAP" }, "image_size": 224, "initializer_range": 0.02, "intermediate_size": 3072, "label2id": { "ASCOT CAP": 8, "BASEBALL CAP": 19, "BERET": 0, "BICORNE": 11, "BOATER": 14, "BOWLER": 17, "DEERSTALKER": 7, "FEDORA": 1, "FEZ": 4, "FOOTBALL HELMET": 12, "GARRISON CAP": 18, "HARD HAT": 3, "MILITARY HELMET": 10, "MOTARBOARD": 13, "PITH HELMET": 15, "PORK PIE": 9, "SOMBERO": 2, "SOUTHWESTER": 16, "TOP HAT": 6, "ZUCCHETTO": 5 }, "layer_norm_eps": 1e-12, "model_type": "vit", "num_attention_heads": 12, "num_channels": 3, "num_hidden_layers": 12, "patch_size": 16, "problem_type": "single_label_classification", "qkv_bias": true, "torch_dtype": "float32", "transformers_version": "4.34.0" }