File size: 1,001 Bytes
e315025 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 |
{
"_name_or_path": "/data/ckpt/converted/vilt_pretrain",
"architectures": [
"ViltForQuestionAnswering"
],
"bos_token_id": 0,
"contrast_method": "none",
"distill": false,
"draw_false_text": 0,
"drop_rate": 0.1,
"eos_token_id": 2,
"freeze_patterns": null,
"hidden_size": 768,
"ib_kl": false,
"initializer_range": 0.02,
"itm_loss": 1,
"keep_ratio": 1,
"max_text_len": 40,
"merge_r": 0,
"merge_ratio": 0,
"merge_style": "tip",
"merge_text": 0,
"mlm_loss": 1,
"mlm_prob": 0.15,
"mlp_ratio": 4,
"model_type": "vilt",
"mpp_loss": 0,
"num_heads": 12,
"num_layers": 12,
"pad_token_id": 1,
"prune_layers": null,
"prune_method": "mlp_states",
"prune_r": 0,
"reduce_layers": null,
"sim_method": "first_head",
"token_types": 2,
"tokenizer": "bert-base-uncased",
"torch_dtype": "float32",
"transformers_version": "4.25.1",
"vit": "vit_base_patch32_384",
"vocab_size": 30522,
"vqa_label_size": 3129,
"whole_word_masking": false
}
|