rajpurkar/squad
Viewer • Updated • 98.2k • 160k • 364
How to use BaoKien/my_awesome_qa_model with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("question-answering", model="BaoKien/my_awesome_qa_model") # Load model directly
from transformers import AutoTokenizer, AutoModelForQuestionAnswering
tokenizer = AutoTokenizer.from_pretrained("BaoKien/my_awesome_qa_model")
model = AutoModelForQuestionAnswering.from_pretrained("BaoKien/my_awesome_qa_model")This model is a fine-tuned version of albert-base-v2 on the squad 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 |
|---|---|---|---|
| 0.889 | 1.0 | 4380 | 0.8729 |
| 0.6755 | 2.0 | 8760 | 0.8529 |
| 0.4681 | 3.0 | 13140 | 0.9515 |