| { | |
| "_name_or_path": "/data/ckpt/converted/vilt_pretrain", | |
| "architectures": [ | |
| "ViltForVisualEntailment" | |
| ], | |
| "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_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", | |
| "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 | |
| } | |