File size: 2,793 Bytes
83115b0 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 |
{
"model_type": "SimpleGenerativeModel",
"base_model_type": "pie_modules.models.base_models.BartAsPointerNetwork",
"base_model_config": {
"pretrained_model_name_or_path": "facebook/bart-base",
"use_encoder_mlp": true,
"use_constraints_encoder_mlp": true,
"decoder_position_id_pattern": [
0,
0,
1,
0,
0,
1,
1
],
"encoder_layer_norm_decay": 0.005,
"lr": 6e-05,
"num_beams": 4,
"max_length": 768,
"decoder_layer_norm_decay": 0.05,
"bos_token_id": 0,
"eos_token_id": 1,
"pad_token_id": 1,
"target_token_ids": [
0,
2,
50267,
50265,
50268,
50270,
50271,
50269,
50272,
50266
],
"embedding_weight_mapping": {
"50267": [
4146
],
"50265": [
3618,
1215,
31628
],
"50268": [
414
],
"50270": [
308,
1215,
31628
],
"50271": [
37820
],
"50269": [
1667,
1215,
1116,
1215,
41690
],
"50272": [
9031,
38600,
1215,
41690
],
"50266": [
4548
]
}
},
"override_generation_kwargs": null,
"warmup_proportion": 0.01,
"learning_rate": null,
"optimizer_type": null,
"metric_stages": [
"val",
"test",
"train"
],
"metric_call_predict": [
"val",
"test"
],
"taskmodule_config": {
"taskmodule_type": "PointerNetworkTaskModuleForEnd2EndRE",
"tokenizer_name_or_path": "facebook/bart-base",
"document_type": "pytorch_ie.documents.TextDocumentWithLabeledSpansBinaryRelationsAndLabeledPartitions",
"tokenized_document_type": "pie_modules.documents.TokenDocumentWithLabeledSpansBinaryRelationsAndLabeledPartitions",
"span_layer_name": "labeled_spans",
"relation_layer_name": "binary_relations",
"none_label": "none",
"loop_dummy_relation_name": "loop",
"constrained_generation": true,
"label_tokens": null,
"label_representations": null,
"labels_per_layer": {
"labeled_spans": [
"background_claim",
"data",
"own_claim"
],
"binary_relations": [
"contradicts",
"parts_of_same",
"semantically_same",
"supports"
]
},
"exclude_labels_per_layer": null,
"create_constraints": true,
"tokenizer_init_kwargs": {
"add_prefix_space": true
},
"tokenizer_kwargs": {
"truncation": true,
"return_overflowing_tokens": true,
"stride": 128,
"strict_span_conversion": false
},
"partition_layer_name": "labeled_partitions",
"annotation_field_mapping": null,
"log_first_n_examples": null
}
} |