vilt_finetuned_fashion / config.json
Ornelas's picture
Training in progress, step 10
4c6e0c9 verified
raw
history blame
1.36 kB
{
"_name_or_path": "dandelin/vilt-b32-mlm",
"architectures": [
"ViltForQuestionAnswering"
],
"attention_probs_dropout_prob": 0.0,
"hidden_act": "gelu",
"hidden_dropout_prob": 0.0,
"hidden_size": 768,
"id2label": {
"0": "Blouses Shirts",
"1": "Jackets Coats",
"2": "Rompers Jumpsuits",
"3": "Shorts",
"4": "Tees Tanks",
"5": "Dresses",
"6": "Pants",
"7": "Shirts Polos",
"8": "Skirts",
"9": "Cardigans",
"10": "Sweaters",
"11": "Sweatshirts Hoodies",
"12": "Graphic Tees"
},
"image_size": 384,
"initializer_range": 0.02,
"intermediate_size": 3072,
"label2id": {
"Blouses Shirts": 0,
"Cardigans": 9,
"Dresses": 5,
"Graphic Tees": 12,
"Jackets Coats": 1,
"Pants": 6,
"Rompers Jumpsuits": 2,
"Shirts Polos": 7,
"Shorts": 3,
"Skirts": 8,
"Sweaters": 10,
"Sweatshirts Hoodies": 11,
"Tees Tanks": 4
},
"layer_norm_eps": 1e-12,
"max_image_length": -1,
"max_position_embeddings": 40,
"modality_type_vocab_size": 2,
"model_type": "vilt",
"num_attention_heads": 12,
"num_channels": 3,
"num_hidden_layers": 12,
"num_images": -1,
"patch_size": 32,
"qkv_bias": true,
"tie_word_embeddings": false,
"torch_dtype": "float32",
"transformers_version": "4.38.2",
"type_vocab_size": 2,
"vocab_size": 30522
}