malayalam-moe-pretrain

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

  • Loss: 1.1966

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: 5e-05
  • train_batch_size: 4
  • eval_batch_size: 4
  • seed: 42
  • gradient_accumulation_steps: 64
  • total_train_batch_size: 256
  • optimizer: Use OptimizerNames.ADAMW_TORCH_FUSED 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: 0.02
  • num_epochs: 12
  • mixed_precision_training: Native AMP

Training results

Training Loss Epoch Step Validation Loss
95.2033 1.2500 500 1.4825
81.1107 2.5000 1000 1.3183
79.8998 3.0 1200 1.3110
78.8520 3.7500 1500 1.2993
72.8643 5.0 2000 1.2758
68.8933 6.0 2400 1.2886
70.5932 6.2500 2500 1.3040
65.6363 7.5000 3000 1.3147
60.5359 8.7500 3500 1.3476
60.8426 9.0 3600 1.3474
79.4108 1.3464 4000 1.2411
77.1449 1.5147 4500 1.1966

Framework versions

  • Transformers 5.16.1
  • Pytorch 2.10.0+cu128
  • Datasets 5.0.1
  • Tokenizers 0.23.1
Downloads last month
1,119
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

Model tree for siyah1/malayalam-moe-pretrain

Unable to build the model tree, the base model loops to the model itself. Learn more.