jinaai
/

jina-bert-flash-implementation

🇪🇺 Region: EU

Model card Files Files and versions

Markus28 commited on Feb 22, 2024

Commit

c4185ce

·

1 Parent(s): a1e1eff

debugging

Files changed (1) hide show

modeling_bert.py +1 -1

modeling_bert.py CHANGED Viewed

@@ -501,7 +501,7 @@ class BertForPreTraining(BertPreTrainedModel):
                 )
         prediction_scores, seq_relationship_score = self.cls(sequence_output, pooled_output)
-        print(labels)
         assert False
         if (

                 )
         prediction_scores, seq_relationship_score = self.cls(sequence_output, pooled_output)
+        print(labels.flatten()[masked_token_idx])
         assert False
         if (