Edit model card

lua-mistral-1L-mini

This model is a mini single-layer Mistral model pre-trained on on the nilq/small-lua-stack dataset. It achieves the following results on the evaluation set:

  • Loss: 3.0245
  • Accuracy: 0.4208

Model description

This model might contain some very simple model of Lua.

Intended uses & limitations

Let's see if we can find some interesting stuff inside this model.

Training and evaluation data

Trained on the Lua subset of The Stack.

Training procedure

Training hyperparameters

The following hyperparameters were used during training:

  • learning_rate: 0.0006
  • train_batch_size: 64
  • eval_batch_size: 8
  • seed: 42
  • optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
  • lr_scheduler_type: cosine
  • num_epochs: 3.0

Training results

  • Loss: 3.016

Framework versions

  • Transformers 4.38.1
  • Pytorch 2.2.0+cu121
  • Datasets 2.17.1
  • Tokenizers 0.15.2
Downloads last month
4
Safetensors
Model size
4.13M params
Tensor type
F32
·

Dataset used to train nilq/lua-mistral-1L-mini

Collection including nilq/lua-mistral-1L-mini

Evaluation results