Markus28 commited on
Commit
c2d8dc3
1 Parent(s): c4185ce

removed debugging

Browse files
Files changed (1) hide show
  1. modeling_bert.py +0 -3
modeling_bert.py CHANGED
@@ -501,9 +501,6 @@ class BertForPreTraining(BertPreTrainedModel):
501
  )
502
  prediction_scores, seq_relationship_score = self.cls(sequence_output, pooled_output)
503
 
504
- print(labels.flatten()[masked_token_idx])
505
- assert False
506
-
507
  if (
508
  self.dense_seq_output and labels is not None
509
  ): # prediction_scores are already flattened
 
501
  )
502
  prediction_scores, seq_relationship_score = self.cls(sequence_output, pooled_output)
503
 
 
 
 
504
  if (
505
  self.dense_seq_output and labels is not None
506
  ): # prediction_scores are already flattened