Update modeling_sim.py
Browse files- modeling_sim.py +2 -0
modeling_sim.py
CHANGED
@@ -13,4 +13,6 @@ class SimModel(MobileBertPreTrainedModel):
|
|
13 |
def forward(self, input_ids, attention_mask, token_type_ids, return_dict):
|
14 |
print(input_ids, attention_mask, token_type_ids)
|
15 |
print(self.word_embeddings(input_ids).size())
|
|
|
|
|
16 |
return (self.word_embeddings(input_ids), None, None)
|
|
|
13 |
def forward(self, input_ids, attention_mask, token_type_ids, return_dict):
|
14 |
print(input_ids, attention_mask, token_type_ids)
|
15 |
print(self.word_embeddings(input_ids).size())
|
16 |
+
import pdb
|
17 |
+
pdb.set_trace()
|
18 |
return (self.word_embeddings(input_ids), None, None)
|