srl_bert / config.json
martincc98's picture
Upload config
7b35f29 verified
{
"_name_or_path": "martincc98/srl_bert",
"activation": "gelu",
"architectures": [
"DistilBertForTokenClassification"
],
"attention_dropout": 0.1,
"dim": 768,
"dropout": 0.1,
"hidden_dim": 3072,
"id2label": {
"0": "ARG0",
"1": "ARG1",
"2": "ARG1-DSP",
"3": "ARG2",
"4": "ARG3",
"5": "ARG4",
"6": "ARG5",
"7": "ARGA",
"8": "ARGM-ADJ",
"9": "ARGM-ADV",
"10": "ARGM-CAU",
"11": "ARGM-COM",
"12": "ARGM-CXN",
"13": "ARGM-DIR",
"14": "ARGM-DIS",
"15": "ARGM-EXT",
"16": "ARGM-GOL",
"17": "ARGM-LOC",
"18": "ARGM-LVB",
"19": "ARGM-MNR",
"20": "ARGM-MOD",
"21": "ARGM-NEG",
"22": "ARGM-PRD",
"23": "ARGM-PRP",
"24": "ARGM-PRR",
"25": "ARGM-REC",
"26": "ARGM-TMP",
"27": "O",
"28": "V"
},
"initializer_range": 0.02,
"label2id": {
"ARG0": 0,
"ARG1": 1,
"ARG1-DSP": 2,
"ARG2": 3,
"ARG3": 4,
"ARG4": 5,
"ARG5": 6,
"ARGA": 7,
"ARGM-ADJ": 8,
"ARGM-ADV": 9,
"ARGM-CAU": 10,
"ARGM-COM": 11,
"ARGM-CXN": 12,
"ARGM-DIR": 13,
"ARGM-DIS": 14,
"ARGM-EXT": 15,
"ARGM-GOL": 16,
"ARGM-LOC": 17,
"ARGM-LVB": 18,
"ARGM-MNR": 19,
"ARGM-MOD": 20,
"ARGM-NEG": 21,
"ARGM-PRD": 22,
"ARGM-PRP": 23,
"ARGM-PRR": 24,
"ARGM-REC": 25,
"ARGM-TMP": 26,
"O": 27,
"V": 28
},
"max_position_embeddings": 512,
"model_type": "distilbert",
"n_heads": 12,
"n_layers": 6,
"pad_token_id": 0,
"qa_dropout": 0.1,
"seq_classif_dropout": 0.2,
"sinusoidal_pos_embds": false,
"tie_weights_": true,
"torch_dtype": "float32",
"transformers_version": "4.39.1",
"vocab_size": 30522
}