yutingg
/

custom-distill-bert-for-sentence-label

Feature Extraction

distill_bert_sentence_label

Model card Files Files and versions Community

yutingg commited on Dec 14, 2023

Commit

3b6adde

·

1 Parent(s): 0a66ae1

Update modeling_sentence_label.py

Files changed (1) hide show

modeling_sentence_label.py +1 -1

modeling_sentence_label.py CHANGED Viewed

@@ -27,4 +27,4 @@ class DistillBERTSentenceLabel(PreTrainedModel):
         output = self.classifier(pooler)
         # output = self.sigmoid(output)
         # output = self.softmax(output)
-        return return {"logits": output}

         output = self.classifier(pooler)
         # output = self.sigmoid(output)
         # output = self.softmax(output)
+        return {"logits": output}