Updated for cnlpt v0.6.0

config.json

@@ -2,6 +2,7 @@
   "architectures": [
     "CnlpModelForClassification"
   ],
+  "cnlpt_version": "0.6.0",
   "encoder_config": {
     "_name_or_path": "microsoft/BiomedNLP-PubMedBERT-base-uncased-abstract",
     "add_cross_attention": false,
@@ -83,74 +84,67 @@
   },
   "encoder_name": "microsoft/BiomedNLP-PubMedBERT-base-uncased-abstract",
   "finetuning_task": [
-    "event",
     "timex",
+    "event",
     "tlinkx"
   ],
   "hidden_dropout_prob": 0.1,
   "hidden_size": 768,
   "hier_head_config": null,
-  "label_dictionary": [
-    {
-      "event": [
-        "B-AFTER",
-        "B-BEFORE",
-        "B-BEFORE/OVERLAP",
-        "B-OVERLAP",
-        "I-AFTER",
-        "I-BEFORE",
-        "I-BEFORE/OVERLAP",
-        "I-OVERLAP",
-        "O"
-      ],
-      "timex": [
-        "B-DATE",
-        "B-DURATION",
-        "B-PREPOSTEXP",
-        "B-QUANTIFIER",
-        "B-SECTIONTIME",
-        "B-SET",
-        "B-TIME",
-        "I-DATE",
-        "I-DURATION",
-        "I-PREPOSTEXP",
-        "I-QUANTIFIER",
-        "I-SECTIONTIME",
-        "I-SET",
-        "I-TIME",
-        "O"
-      ],
-      "tlinkx": [
-        "BEFORE",
-        "BEGINS-ON",
-        "CONTAINS",
-        "CONTAINS-SUBEVENT",
-        "ENDS-ON",
-        "NOTED-ON",
-        "None",
-        "OVERLAP"
-      ]
-    }
-  ],
-  "layer": 10,
+  "label_dictionary": {
+    "event": [
+      "B-AFTER",
+      "B-BEFORE",
+      "B-BEFORE/OVERLAP",
+      "B-OVERLAP",
+      "I-AFTER",
+      "I-BEFORE",
+      "I-BEFORE/OVERLAP",
+      "I-OVERLAP",
+      "O"
+    ],
+    "timex": [
+      "B-DATE",
+      "B-DURATION",
+      "B-PREPOSTEXP",
+      "B-QUANTIFIER",
+      "B-SECTIONTIME",
+      "B-SET",
+      "B-TIME",
+      "I-DATE",
+      "I-DURATION",
+      "I-PREPOSTEXP",
+      "I-QUANTIFIER",
+      "I-SECTIONTIME",
+      "I-SET",
+      "I-TIME",
+      "O"
+    ],
+    "tlinkx": [
+      "BEFORE",
+      "BEGINS-ON",
+      "CONTAINS",
+      "CONTAINS-SUBEVENT",
+      "ENDS-ON",
+      "NOTED-ON",
+      "None",
+      "OVERLAP"
+    ]
+  },
+  "layer": 11,
   "model_type": "cnlpt",
-  "num_labels_list": [
-    9,
-    15,
-    8
-  ],
   "num_rel_attention_heads": 256,
   "rel_attention_head_dims": 64,
-  "relations": [
-    false,
-    false,
-    true
-  ],
-  "tagger": [
-    true,
-    true,
-    false
-  ],
+  "relations": {
+    "event": false,
+    "timex": false,
+    "tlinkx": true
+  },
+  "tagger": {
+    "event": true,
+    "timex": true,
+    "tlinkx": false
+  },
   "tokens": false,
   "torch_dtype": "float32",
   "transformers_version": "4.22.2",

pytorch_model.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:9daf34b2db77c26043e923b13458d97555e55d3afcd3ffde55ffef3683555825
-size 538633829
+oid sha256:3fe182dff2f790f67db9eb8c87138ed4fb9ccfe368a9bcb37585d3483a395abe
+size 538633893