{ | |
"_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": "Skirt", | |
"1": "Jodhpurs", | |
"10": "Leggings", | |
"11": "Coverup", | |
"12": "Capris", | |
"13": "Dress", | |
"14": "Sweatshorts", | |
"15": "Tee", | |
"16": "Jersey", | |
"17": "Cutoffs", | |
"18": "Sweatpants", | |
"19": "Sarong", | |
"2": "Henley", | |
"20": "Joggers", | |
"21": "Tank", | |
"22": "Poncho", | |
"23": "Onesie", | |
"24": "Anorak", | |
"25": "Kimono", | |
"26": "Romper", | |
"27": "Halter", | |
"28": "Top", | |
"29": "Culottes", | |
"3": "Jeggings", | |
"30": "Robe", | |
"31": "Shorts", | |
"32": "Jeans", | |
"33": "Cardigan", | |
"34": "Jumpsuit", | |
"35": "Sweater", | |
"36": "Bomber", | |
"37": "Parka", | |
"38": "Chinos", | |
"39": "Turtleneck", | |
"4": "Flannel", | |
"40": "Trunks", | |
"41": "Blouse", | |
"42": "Gauchos", | |
"43": "Blazer", | |
"44": "Caftan", | |
"45": "Hoodie", | |
"5": "Coat", | |
"6": "Peacoat", | |
"7": "Jacket", | |
"8": "Button-Down", | |
"9": "Kaftan" | |
}, | |
"image_size": 224, | |
"initializer_range": 0.02, | |
"intermediate_size": 3072, | |
"label2id": { | |
"Anorak": "24", | |
"Blazer": "43", | |
"Blouse": "41", | |
"Bomber": "36", | |
"Button-Down": "8", | |
"Caftan": "44", | |
"Capris": "12", | |
"Cardigan": "33", | |
"Chinos": "38", | |
"Coat": "5", | |
"Coverup": "11", | |
"Culottes": "29", | |
"Cutoffs": "17", | |
"Dress": "13", | |
"Flannel": "4", | |
"Gauchos": "42", | |
"Halter": "27", | |
"Henley": "2", | |
"Hoodie": "45", | |
"Jacket": "7", | |
"Jeans": "32", | |
"Jeggings": "3", | |
"Jersey": "16", | |
"Jodhpurs": "1", | |
"Joggers": "20", | |
"Jumpsuit": "34", | |
"Kaftan": "9", | |
"Kimono": "25", | |
"Leggings": "10", | |
"Onesie": "23", | |
"Parka": "37", | |
"Peacoat": "6", | |
"Poncho": "22", | |
"Robe": "30", | |
"Romper": "26", | |
"Sarong": "19", | |
"Shorts": "31", | |
"Skirt": "0", | |
"Sweater": "35", | |
"Sweatpants": "18", | |
"Sweatshorts": "14", | |
"Tank": "21", | |
"Tee": "15", | |
"Top": "28", | |
"Trunks": "40", | |
"Turtleneck": "39" | |
}, | |
"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.29.2" | |
} | |