metadata
license: cc-by-4.0
pythia-1.4B-finetuned-oa-instructions
This model is a fine-tuned version of pythia on the oa dataset. It achieves the following results on the evaluation set:
Loss: 0.1224
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:
seed: 42
learning_rate: 5e-06
train_batch_size: 32
eval_batch_size: 8
optimizer: Adam with betas : {'lr': 5e-06, 'betas': [0.9, 0.999], 'eps': 1e-08, 'weight_decay': 0.0}
lr_scheduler_type: linear
training_steps: 5000
fp16
warmup_steps 5
Num examples = 53k
Training results
{
"epoch": 1.0,
"train_loss": 0.8031303182039198,
"train_runtime": 6338.6403,
"train_samples": 53455,
"train_samples_per_second": 8.433,
"train_steps_per_second": 0.264
}
Framework versions
- transformers 4.24.0
- torch 1.10.0+cu111
- datasets 2.10.0
- tokenizers 0.12.1