Hugging Face's logo

zyacub
/
hugging_face_QA

Question Answering
Transformers
PyTorch
TensorBoard
distilbert
Generated from Trainer
Inference Endpoints
Model card Files Metrics Community
1
hugging_face_QA / README.md
zyacub's picture
zyacub
update model card README.md
b2c848f
preview code
|
raw
history blame contribute delete
1.52 kB
metadata
license: apache-2.0
base_model: distilbert-base-uncased
tags:
  - generated_from_trainer
datasets:
  - squad
model-index:
  - name: hugging_face_QA
    results: []

hugging_face_QA

This model is a fine-tuned version of distilbert-base-uncased on the squad dataset. It achieves the following results on the evaluation set:

  • Loss: 2.1866

Model description

More information needed

Intended uses & limitations

More information needed

Training and evaluation data

More information needed

Training procedure

Training hyperparameters

The following hyperparameters were used during training:

  • learning_rate: 2e-05
  • train_batch_size: 16
  • eval_batch_size: 16
  • seed: 42
  • optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
  • lr_scheduler_type: linear
  • num_epochs: 5

Training results

Training Loss Epoch Step Validation Loss
No log 1.0 250 2.0354
0.3317 2.0 500 2.1035
0.3317 3.0 750 2.0138
0.313 4.0 1000 2.1020
0.313 5.0 1250 2.1866

Framework versions

  • Transformers 4.31.0
  • Pytorch 2.0.1+cu118
  • Datasets 2.14.4
  • Tokenizers 0.13.3