| { | |
| "text_encoder": { | |
| "model_type": "bert", | |
| "dim": 384, | |
| "context_dim": 768, | |
| "vocab_size": 250037, | |
| "padding_idx": 1, | |
| "num_layers": 12, | |
| "num_heads": 12, | |
| "embedding_dim": 256, | |
| "multimodal_layers_ids": [8, 9, 10, 11], | |
| "head_one_neuron": true, | |
| "pooling": "mean", | |
| "max_position_embeddings": 77, | |
| "dropout_prob": 0.1 | |
| }, | |
| "image_encoder": { | |
| "dim": 768, | |
| "patch_size": 16, | |
| "image_size": 224, | |
| "num_layers": 12, | |
| "num_heads": 12, | |
| "embedding_dim": 256, | |
| "pooling": "cls" | |
| } | |
| } |