nyu-mll/glue
Viewer • Updated • 1.49M • 472k • 504
How to use TransferGraph/Capreolus_bert-base-msmarco-finetuned-lora-glue_cola with PEFT:
from peft import PeftModel
from transformers import AutoModelForSequenceClassification
base_model = AutoModelForSequenceClassification.from_pretrained("Capreolus/bert-base-msmarco")
model = PeftModel.from_pretrained(base_model, "TransferGraph/Capreolus_bert-base-msmarco-finetuned-lora-glue_cola")This model is a fine-tuned version of Capreolus/bert-base-msmarco on the glue 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:
| accuracy | train_loss | epoch |
|---|---|---|
| 0.3087 | None | 0 |
| 0.6922 | 0.7929 | 0 |
| 0.7335 | 0.5786 | 1 |
| 0.7574 | 0.5167 | 2 |
| 0.7747 | 0.4915 | 3 |
Base model
Capreolus/bert-base-msmarco