Update BERT.py

CSgaoan · Mar 25, 2023 · 1185af1 · 1185af1
1 parent 419b654
commit 1185af1
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/model/BERT.py b/model/BERT.py
@@ -377,11 +377,11 @@ def forward(self, x, mask_cls):
         return sent_scores
 
 
-class ExtSummarizer(nn.Module):
+class ExtSummarizer_F(nn.Module):
 
     def __init__(self):
 
-        super(ExtSummarizer, self).__init__()
+        super(ExtSummarizer_F, self).__init__()
 
         self.bert = BertModel.from_pretrained('bert-base-uncased')