Update models.py

Added few fixed parameters

models.py

@@ -19,11 +19,12 @@ class BertPooler(nn.Module):
         return pooled_output
 
 
+
 class Model_Rational_Label(BertPreTrainedModel):
-     def __init__(self,config,params):
+     def __init__(self,config):
         super().__init__(config)
-        self.num_labels=params['num_classes']
-        self.impact_factor=params['rationale_impact']
+        self.num_labels=2
+        self.impact_factor=0.8
         self.bert = BertModel(config,add_pooling_layer=False)
         self.bert_pooler=BertPooler(config)
         self.token_dropout = nn.Dropout(0.1)
@@ -33,8 +34,8 @@ class Model_Rational_Label(BertPreTrainedModel):
         self.init_weights()        
 #         self.embeddings = AutoModelForTokenClassification.from_pretrained(params['model_path'], cache_dir=params['cache_path'])
         
-     def forward(self, input_ids=None, mask=None, attn=None, labels=None):
-        outputs = self.bert(input_ids, mask)
+     def forward(self, input_ids=None, attention_mask=None, token_type_ids=None, attn=None, labels=None):
+        outputs = self.bert(input_ids, attention_mask)
         # out = outputs.last_hidden_state
         out=outputs[0]
         logits = self.token_classifier(self.token_dropout(out))