Mariofm02's picture
Model save
a587258 verified
{
"_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": "ace of clubs",
"1": "ace of diamonds",
"10": "five of hearts",
"11": "five of spades",
"12": "four of clubs",
"13": "four of diamonds",
"14": "four of hearts",
"15": "four of spades",
"16": "jack of clubs",
"17": "jack of diamonds",
"18": "jack of hearts",
"19": "jack of spades",
"2": "ace of hearts",
"20": "king of clubs",
"21": "king of diamonds",
"22": "king of hearts",
"23": "king of spades",
"24": "nine of clubs",
"25": "nine of diamonds",
"26": "nine of hearts",
"27": "nine of spades",
"28": "queen of clubs",
"29": "queen of diamonds",
"3": "ace of spades",
"30": "queen of hearts",
"31": "queen of spades",
"32": "seven of clubs",
"33": "seven of diamonds",
"34": "seven of hearts",
"35": "seven of spades",
"36": "six of clubs",
"37": "six of diamonds",
"38": "six of hearts",
"39": "six of spades",
"4": "eight of clubs",
"40": "ten of clubs",
"41": "ten of diamonds",
"42": "ten of hearts",
"43": "ten of spades",
"44": "three of clubs",
"45": "three of diamonds",
"46": "three of hearts",
"47": "three of spades",
"48": "two of clubs",
"49": "two of diamonds",
"5": "eight of diamonds",
"50": "two of hearts",
"51": "two of spades",
"6": "eight of hearts",
"7": "eight of spades",
"8": "five of clubs",
"9": "five of diamonds"
},
"image_size": 224,
"initializer_range": 0.02,
"intermediate_size": 3072,
"label2id": {
"ace of clubs": "0",
"ace of diamonds": "1",
"ace of hearts": "2",
"ace of spades": "3",
"eight of clubs": "4",
"eight of diamonds": "5",
"eight of hearts": "6",
"eight of spades": "7",
"five of clubs": "8",
"five of diamonds": "9",
"five of hearts": "10",
"five of spades": "11",
"four of clubs": "12",
"four of diamonds": "13",
"four of hearts": "14",
"four of spades": "15",
"jack of clubs": "16",
"jack of diamonds": "17",
"jack of hearts": "18",
"jack of spades": "19",
"king of clubs": "20",
"king of diamonds": "21",
"king of hearts": "22",
"king of spades": "23",
"nine of clubs": "24",
"nine of diamonds": "25",
"nine of hearts": "26",
"nine of spades": "27",
"queen of clubs": "28",
"queen of diamonds": "29",
"queen of hearts": "30",
"queen of spades": "31",
"seven of clubs": "32",
"seven of diamonds": "33",
"seven of hearts": "34",
"seven of spades": "35",
"six of clubs": "36",
"six of diamonds": "37",
"six of hearts": "38",
"six of spades": "39",
"ten of clubs": "40",
"ten of diamonds": "41",
"ten of hearts": "42",
"ten of spades": "43",
"three of clubs": "44",
"three of diamonds": "45",
"three of hearts": "46",
"three of spades": "47",
"two of clubs": "48",
"two of diamonds": "49",
"two of hearts": "50",
"two of spades": "51"
},
"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.38.2"
}