CoolFace
Modelpublic

yahyaahmed/tinyllama-lora-squad_16_5e-04_2_lora16_qkv

sourceHugging Faceapache-2.0updated 1y agoView on Hugging Face
0likes19downloads
Model Card

<!-- This model card has been generated automatically according to the information the Trainer had access to. You should probably proofread and complete it, then remove this comment. -->

tinyllama-lora-squad165e-042lora16_qkv

This model is a fine-tuned version of TinyLlama/TinyLlama-1.1B-Chat-v1.0 on an unknown dataset. It achieves the following results on the evaluation set:

  • —Loss: 0.0045

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.0005
  • —trainbatchsize: 4
  • —evalbatchsize: 4
  • —seed: 42
  • —gradientaccumulationsteps: 4
  • —totaltrainbatch_size: 16
  • —optimizer: Use OptimizerNames.ADAMWTORCH with betas=(0.9,0.999) and epsilon=1e-08 and optimizerargs=No additional optimizer arguments
  • —lrschedulertype: linear
  • —lrschedulerwarmup_steps: 100
  • —num_epochs: 2
  • —mixedprecisiontraining: Native AMP

Training results

Training LossEpochStepValidation Loss
2.02850.19952170.0079
0.00550.39904340.0061
0.00470.59856510.0053
0.00430.79808680.0049
0.00430.997510850.0048
0.00341.196713020.0048
0.00331.396215190.0046
0.00311.595717360.0047
0.00311.795219530.0045
0.00321.994721700.0045

Framework versions

  • —PEFT 0.15.2
  • —Transformers 4.52.4
  • —Pytorch 2.6.0+cu124
  • —Datasets 3.6.0
  • —Tokenizers 0.21.1