jwieting
/

paraphrastic_test

Sentence Similarity

sentence-transformers

feature-extraction

text-embeddings-inference

Model card Files Files and versions

jwieting commited on Nov 14, 2022

Commit

a7d623d

·

1 Parent(s): 6dd5977

Update modeling_paragram_sp.py

Files changed (1) hide show

modeling_paragram_sp.py +2 -2

modeling_paragram_sp.py CHANGED Viewed

@@ -11,8 +11,8 @@ class ParagramSPModel(BertPreTrainedModel):
         self.post_init()
     def forward(self, input_ids, attention_mask):
-        #print(input_ids)
-        #print(attention_mask)
         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)

         self.post_init()
     def forward(self, input_ids, attention_mask):
+        print(input_ids)
+        print(attention_mask)
         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)