Shannon Shen
add model
543455c
{
"architectures": [
"HierarchicalModelForTokenClassification"
],
"finetuning_task": "token_classification",
"hidden_dropout_prob": 0.1,
"hidden_size": 768,
"id2label": {
"0": "BIB_INFO",
"1": "REFERENCES",
"2": "UNKNOWN",
"3": "BODY_CONTENT",
"4": "PAGE_NUMBER",
"5": "TABLE",
"6": "ACKNOWLEDGMENT",
"7": "FIGURE",
"8": "CONFLICT_STATEMENT",
"9": "AFFILIATION",
"10": "DATES",
"11": "TYPE",
"12": "TITLE",
"13": "AUTHOR",
"14": "ABSTRACT",
"15": "CORRESPONDENCE",
"16": "EDITOR",
"17": "COPYRIGHT",
"18": "AUTHOR_TITLE",
"19": "KEYWORDS",
"20": "GLOSSARY",
"21": "EQUATION"
},
"initializer_range": 0.02,
"label2id": {
"ABSTRACT": "14",
"ACKNOWLEDGMENT": "6",
"AFFILIATION": "9",
"AUTHOR": "13",
"AUTHOR_TITLE": "18",
"BIB_INFO": "0",
"BODY_CONTENT": "3",
"CONFLICT_STATEMENT": "8",
"COPYRIGHT": "17",
"CORRESPONDENCE": "15",
"DATES": "10",
"EDITOR": "16",
"EQUATION": "21",
"FIGURE": "7",
"GLOSSARY": "20",
"KEYWORDS": "19",
"PAGE_NUMBER": "4",
"REFERENCES": "1",
"TABLE": "5",
"TITLE": "12",
"TYPE": "11",
"UNKNOWN": "2"
},
"load_weights_from_existing_model": false,
"model_type": "hierarchical_model",
"pad_token_id": 0,
"textline_encoder_output": "average",
"textline_encoder_type": "bert-layer",
"textline_encoder_used_bert_layer": "first",
"textline_model_type": "layoutlm-base-uncased",
"textline_model_used_bert_layer": "first",
"transformers_version": "4.7.0",
"vila_preprocessor_config": {
"agg_level": "block",
"group_bbox_agg": "first"
},
"vocab_size": 30522
}