Update modeling_paragram_sp.py
Browse files- modeling_paragram_sp.py +2 -0
modeling_paragram_sp.py
CHANGED
|
@@ -29,6 +29,8 @@ class ParagramSPModel(BertPreTrainedModel):
|
|
| 29 |
print(attention_mask)
|
| 30 |
input_ids = self.filter_input_ids(input_ids)
|
| 31 |
attention_mask = input_ids != self.config.pad_token_id
|
|
|
|
|
|
|
| 32 |
embeddings = self.word_embeddings(input_ids)
|
| 33 |
masked_embeddings = embeddings * attention_mask[:, :, None]
|
| 34 |
mean_pooled_embeddings = masked_embeddings.sum(dim=1) / attention_mask[:, :, None].sum(dim=1)
|
|
|
|
| 29 |
print(attention_mask)
|
| 30 |
input_ids = self.filter_input_ids(input_ids)
|
| 31 |
attention_mask = input_ids != self.config.pad_token_id
|
| 32 |
+
print(input_ids)
|
| 33 |
+
print(attention_mask)
|
| 34 |
embeddings = self.word_embeddings(input_ids)
|
| 35 |
masked_embeddings = embeddings * attention_mask[:, :, None]
|
| 36 |
mean_pooled_embeddings = masked_embeddings.sum(dim=1) / attention_mask[:, :, None].sum(dim=1)
|