Update modeling_paragram_sp.py

modeling_paragram_sp.py

@@ -10,11 +10,13 @@ class ParagramSPModel(BertPreTrainedModel):
         # Initialize weights and apply final processing
         self.post_init()
 
-    def forward(self, input_ids, attention_mask, token_type_ids, return_dict):
-        print(input_ids)
-        print(attention_mask)
-        print(token_type_ids)
-        print(return_dict)
+    #def forward(self, input_ids, attention_mask, token_type_ids, return_dict):
+    def forward(self, inputs**):
+        print(inputs)
+        #print(input_ids)
+        #print(attention_mask)
+        #print(token_type_ids)
+        #print(return_dict)
         embeddings = self.word_embeddings(input_ids)
         masked_embeddings = embeddings * attention_mask[:, :, None]
         mean_pooled_embeddings = masked_embeddings.sum(dim=1) / attention_mask[:, :, None].sum(dim=1)