nyu-mll/glue
Viewer • Updated • 1.49M • 428k • 523
How to use gokuls/distilbert_sa_GLUE_Experiment_logit_kd_data_aug_cola with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("text-classification", model="gokuls/distilbert_sa_GLUE_Experiment_logit_kd_data_aug_cola") # Load model directly
from transformers import AutoTokenizer, AutoModelForSequenceClassification
tokenizer = AutoTokenizer.from_pretrained("gokuls/distilbert_sa_GLUE_Experiment_logit_kd_data_aug_cola")
model = AutoModelForSequenceClassification.from_pretrained("gokuls/distilbert_sa_GLUE_Experiment_logit_kd_data_aug_cola", device_map="auto")This model is a fine-tuned version of distilbert-base-uncased on the GLUE COLA dataset. It achieves the following results on the evaluation set:
More information needed
More information needed
More information needed
The following hyperparameters were used during training:
| Training Loss | Epoch | Step | Validation Loss | Matthews Correlation |
|---|---|---|---|---|
| 0.5745 | 1.0 | 835 | 0.7299 | 0.1224 |
| 0.3736 | 2.0 | 1670 | 0.7628 | 0.1626 |
| 0.2919 | 3.0 | 2505 | 0.7388 | 0.1954 |
| 0.2517 | 4.0 | 3340 | 0.7483 | 0.1699 |
| 0.2279 | 5.0 | 4175 | 0.7558 | 0.1651 |
| 0.2108 | 6.0 | 5010 | 0.7734 | 0.1542 |