config.json
794 B · 32 lines · json Raw
1 {
2 "_name_or_path": "cross-encoder/ms-marco-MiniLM-L-12-v2",
3 "architectures": [
4 "BertForSequenceClassification"
5 ],
6 "attention_probs_dropout_prob": 0.1,
7 "gradient_checkpointing": false,
8 "hidden_act": "gelu",
9 "hidden_dropout_prob": 0.1,
10 "hidden_size": 384,
11 "id2label": {
12 "0": "LABEL_0"
13 },
14 "initializer_range": 0.02,
15 "intermediate_size": 1536,
16 "label2id": {
17 "LABEL_0": 0
18 },
19 "layer_norm_eps": 1e-12,
20 "max_position_embeddings": 512,
21 "model_type": "bert",
22 "num_attention_heads": 12,
23 "num_hidden_layers": 6,
24 "pad_token_id": 0,
25 "position_embedding_type": "absolute",
26 "transformers_version": "4.4.2",
27 "type_vocab_size": 2,
28 "use_cache": true,
29 "vocab_size": 30522,
30 "sbert_ce_default_activation_function": "torch.nn.modules.linear.Identity"
31 }
32