distilbert
/

distilbert-base-uncased-finetuned-sst-2-english

Text Classification

Eval Results (legacy)

Model card Files Files and versions

Where can I find the prediction head training code?

#10

by ethanjyx - opened Sep 23, 2022

My understanding is this model is using BERT as the backbone and then have a prediction head for the text classification card.

Where can I find the prediction head code? I didn't see it in the "Files and Versions" tab.

Thanks!

Do you mean this https://github.com/huggingface/transformers/blob/71fc33174664738d8c8d93025ebc810180e69c20/src/transformers/models/distilbert/modeling_distilbert.py#L695 ?

Upload images, audio, and videos by dragging in the text input, pasting, or clicking here.

Tap or paste here to upload images

· Sign up or log in to comment