Shideh Almasian
add model
3399294
raw history blame
No virus
1.26 kB
{
"_name_or_path": "/export/data/salmasia/numbert/fine_tune/bert_crf_tagging_8epochs/bert_crf_tagging_seed_19",
"architectures": [
"BERT_CRF_NER"
],
"attention_probs_dropout_prob": 0.1,
"batch_size": 8,
"device": "cuda",
"gradient_checkpointing": false,
"hidden_act": "gelu",
"hidden_dropout_prob": 0.1,
"hidden_size": 768,
"id2label": {
"0": "[CLS]",
"1": "[SEP]",
"2": "[PAD]",
"3": "B-DATE",
"4": "B-DURATION",
"5": "B-SET",
"6": "B-TIME",
"7": "I-DATE",
"8": "I-DURATION",
"9": "I-SET",
"10": "I-TIME",
"11": "O"
},
"initializer_range": 0.02,
"intermediate_size": 3072,
"label2id": {
"B-DATE": 3,
"B-DURATION": 4,
"B-SET": 5,
"B-TIME": 6,
"I-DATE": 7,
"I-DURATION": 8,
"I-SET": 9,
"I-TIME": 10,
"O": 11,
"[CLS]": 0,
"[PAD]": 2,
"[SEP]": 1
},
"layer_norm_eps": 1e-12,
"max_position_embeddings": 512,
"model_type": "bert",
"num_attention_heads": 12,
"num_classes": 12,
"num_hidden_layers": 12,
"pad_token_id": 0,
"position_embedding_type": "absolute",
"start_label_id": 0,
"stop_label_id": 1,
"transformers_version": "4.7.0.dev0",
"type_vocab_size": 2,
"use_cache": true,
"vocab_size": 30522
}