[#42] Refactor training to use the new modeling

aisingapore · Feb 28, 2022 · 7f19533 · 7f19533
1 parent 04b5212
commit 7f19533
Show file tree

Hide file tree

Showing 3 changed files with 673 additions and 664 deletions.
diff --git a/sgnlp/models/dual_bert/modeling.py b/sgnlp/models/dual_bert/modeling.py
@@ -378,8 +378,8 @@ def __init__(self, config,
     def init_bert(self):
         self.bert = BertModel.from_pretrained("bert-base-uncased")
 
-    def forward(self, input_ids_buckets, input_mask_buckets, segment_ids_buckets, input_mask, stance_label_ids,
-                rumor_label_id, stance_position, stance_label_mask, rumor_labels=None, stance_labels=None):
+    def forward(self, input_ids_buckets, segment_ids_buckets, input_mask_buckets, input_mask, stance_position,
+                stance_label_mask, stance_label_ids=None, rumor_label_ids=None, rumor_labels=None, stance_labels=None):
         # def forward(self, input_ids1, token_type_ids1, attention_mask1, input_ids2, token_type_ids2, attention_mask2,
         #           input_ids3, token_type_ids3, attention_mask3, input_ids4, token_type_ids4, attention_mask4,
         #          attention_mask, rumor_labels=None, stance_labels=None, stance_label_mask=None):