llama32-wikitext2-distilled

This model is a fine-tuned version of on an unknown dataset. It achieves the following results on the evaluation set:

  • Loss: 3.8552

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: 0.0001
  • train_batch_size: 4
  • eval_batch_size: 4
  • seed: 42
  • gradient_accumulation_steps: 2
  • total_train_batch_size: 8
  • optimizer: Use OptimizerNames.ADAMW_TORCH with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
  • lr_scheduler_type: linear
  • lr_scheduler_warmup_steps: 500
  • num_epochs: 3
  • mixed_precision_training: Native AMP

Training results

Training Loss Epoch Step Validation Loss
10.1102 0.2619 500 4.9999
9.0742 0.5238 1000 4.5735
8.5261 0.7858 1500 4.3353
7.9589 1.0477 2000 4.2118
7.8108 1.3096 2500 4.1115
7.5611 1.5715 3000 4.0296
7.3756 1.8334 3500 3.9661
6.9172 2.0953 4000 3.9215
6.8475 2.3573 4500 3.8951
6.5798 2.6192 5000 3.8708
6.6206 2.8811 5500 3.8552

Framework versions

  • Transformers 4.50.3
  • Pytorch 2.7.0+cu126
  • Datasets 3.6.0
  • Tokenizers 0.21.1
Downloads last month
6
Safetensors
Model size
0.4B params
Tensor type
F32
ยท
Inference Providers NEW
This model isn't deployed by any Inference Provider. ๐Ÿ™‹ Ask for provider support