removed debugging
Browse files- 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
|