jwieting commited on
Commit
4b4d8b5
·
1 Parent(s): cddb7fe

Update modeling_sim.py

Browse files
Files changed (1) hide show
  1. 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)