stanfordnlp/sst2
Viewer • Updated • 70k • 26.3k • 161
How to use DornierDo17/MiniRoBERTa_SST2 with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("text-classification", model="DornierDo17/MiniRoBERTa_SST2") # Load model directly
from transformers import AutoModel
model = AutoModel.from_pretrained("DornierDo17/MiniRoBERTa_SST2", dtype="auto")This model is a fine-tuned version of my base MiniRoBERTa (17.7M parameters) model. The goal of this fine-tuning experiment was to demonstrate that a RoBERTa-style model, built entirely from scratch and trained on a single GPU with limited compute, can still learn meaningful patterns and adapt effectively to downstream tasks.
The model was fine-tuned on the SST-2 sentiment classification dataset and achieved an accuracy of 80%, which is a strong result given the scale and simplicity of the pretraining setup.
This validates that the model has learned generalizable representations and can be adapted successfully to real-world NLP tasks through fine-tuning.
Base model
DornierDo17/RoBERTa_17.7M