jaandoui
/

DNABERT2-AttentionExtracted

Inference Endpoints

Model card Files Files and versions Community

jaandoui commited on May 14

Commit

7df9fc1

•

1 Parent(s): cefddf8

Update bert_layers.py

Files changed (1) hide show

bert_layers.py +0 -1

bert_layers.py CHANGED Viewed

@@ -940,7 +940,6 @@ class BertForSequenceClassification(BertPreTrainedModel):
             output = (logits,) + outputs[2:]
             return ((loss,) + output) if loss is not None else output
-        print(len(outputs))
         # print(outputs.attentions)
         return SequenceClassifierOutput(

             output = (logits,) + outputs[2:]
             return ((loss,) + output) if loss is not None else output
         # print(outputs.attentions)
         return SequenceClassifierOutput(