1.15.6

yangheng95 · Jun 30, 2022 · 5d66637 · 5d66637
1 parent 37b6d68
commit 5d66637
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 18 deletions.
diff --git a/pyabsa/core/apc/models/fast_lsa_t_v2.py b/pyabsa/core/apc/models/fast_lsa_t_v2.py
@@ -69,23 +69,17 @@ def forward(self, inputs):
             sent_out = self.post_linear(sent_out)
 
         elif self.opt.lcf == 'fusion':
-            # cdw_sent_out = self.CDW_LSA(global_context_features,
-            #                             spc_mask_vec=spc_mask_vec,
-            #                             lcf_matrix=lcf_cdw_matrix,
-            #                             left_lcf_matrix=left_lcf_cdw_matrix,
-            #                             right_lcf_matrix=right_lcf_cdw_matrix)
-            # cdm_sent_out = self.CDM_LSA(global_context_features,
-            #                             spc_mask_vec=spc_mask_vec,
-            #                             lcf_matrix=lcf_cdm_matrix,
-            #                             left_lcf_matrix=left_lcf_cdm_matrix,
-            #                             right_lcf_matrix=right_lcf_cdm_matrix)
-            # sent_out = self.fusion_linear(torch.cat((global_context_features, cdw_sent_out, cdm_sent_out), -1))
-            sent_out = self.CDW_LSA(global_context_features,
-                                    spc_mask_vec=spc_mask_vec,
-                                    lcf_matrix=lcf_cdw_matrix,
-                                    left_lcf_matrix=left_lcf_cdm_matrix,
-                                    right_lcf_matrix=right_lcf_cdm_matrix)
-            sent_out = torch.cat((global_context_features, sent_out), -1)
+            cdw_sent_out = self.CDW_LSA(global_context_features,
+                                        spc_mask_vec=spc_mask_vec,
+                                        lcf_matrix=lcf_cdw_matrix,
+                                        left_lcf_matrix=left_lcf_cdw_matrix,
+                                        right_lcf_matrix=right_lcf_cdw_matrix)
+            cdm_sent_out = self.CDM_LSA(global_context_features,
+                                        spc_mask_vec=spc_mask_vec,
+                                        lcf_matrix=lcf_cdm_matrix,
+                                        left_lcf_matrix=left_lcf_cdm_matrix,
+                                        right_lcf_matrix=right_lcf_cdm_matrix)
+            sent_out = self.fusion_linear(torch.cat((global_context_features, cdw_sent_out, cdm_sent_out), -1))
             sent_out = self.post_linear(sent_out)
 
         else:

diff --git a/release-note.json b/release-note.json
@@ -2,7 +2,10 @@
   "1.15.6": {
     "1": "Add classification report (including precision, recall, F1) display after training",
     "2": "Add offline option to load huggingface model in inference: set get_xxx(offline=True) to auto detect and load local pretrained model",
-    "3": "Add confusion matrix display after training"
+    "3": "Add confusion matrix display after training",
+    "4": "Fix lcf-fusion in fast-lsa-t-v2",
+    "5": "Fix some typos",
+    "6": "Rename MOOC-En dataset to MOOC_En"
   },
   "1.15.5": {
     "1": "Minor fixes of tad inference"