base_model: Qwen/Qwen3-14B
# Automatically upload checkpoint and final model to HF
hub_model_id: Rexhaif/Qwen3-14B-MTEval-SFT
hub_private_repo: false


load_in_8bit: false
load_in_4bit: false
strict: false

chat_template: tokenizer_default
datasets:
  - path: Rexhaif/wmt23-pairs-sft
    split: "train"
    type: chat_template
    field_messages: messages
    roles_to_train: ["assistant"]

shuffle_merged_datasets: true

skip_prepare_dataset: false
dataset_prepared_path: ./data/wmt23-pairs-sft
output_dir: /hnvme/workspace/v106be28-outputs/sft-14b

dataloader_prefetch_factor: 32
dataloader_num_workers: 2
dataloader_pin_memory: true

gc_steps: 1

sequence_len: 512
sample_packing: false
eval_sample_packing: false
pad_to_sequence_len: false

wandb_project: llm-reasoning-mt-eval
wandb_entity:
wandb_name: qwen3-14b-sft

plugins:
  - axolotl.integrations.liger.LigerPlugin
liger_rope: true
liger_rms_norm: true
liger_glu_activation: true
liger_layer_norm: true
liger_fused_linear_cross_entropy: true
gradient_accumulation_steps: 8
micro_batch_size: 8  # should match num_generations / num_gpus

optimizer: adamw_torch_fused
lr_scheduler: cosine
learning_rate: 5.0e-5
cosine_min_lr_ratio: 1.0e-7
max_grad_norm: 1.0
weight_decay: 0.1

bf16: true
tf32: true

flash_attention: true
flash_attn_fuse_qkv: true
flash_attn_fuse_mlp: true
auto_resume_from_checkpoints: true

n_epochs: 3
logging_steps: 10
warmup_ratio: 0.1
evals_per_epoch: 10
saves_per_epoch: 10
save_total_limit: 1
#max_steps: 5000
seed: 42
val_set_size: 0.01

gradient_checkpointing: false
gradient_checkpointing_kwargs:
  use_reentrant: false

Qwen3-14B-MTEval-SFT

This model is a fine-tuned version of Qwen/Qwen3-14B on the Rexhaif/wmt23-pairs-sft dataset. It achieves the following results on the evaluation set:

Loss: 0.2252

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: 8
eval_batch_size: 8
seed: 42
distributed_type: multi-GPU
num_devices: 32
gradient_accumulation_steps: 8
total_train_batch_size: 2048
total_eval_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: cosine
lr_scheduler_warmup_steps: 12
num_epochs: 1.0

Training results

Training Loss	Epoch	Step	Validation Loss
No log	0.0079	1	10.8276
2.6592	0.1023	13	8.0970
3.6616	0.2045	26	0.4104
0.573	0.3068	39	0.3470
0.3716	0.4090	52	0.3575
0.3536	0.5113	65	0.3468
0.3456	0.6136	78	0.3354
0.3213	0.7158	91	0.3314
0.3137	0.8181	104	0.2673
0.2552	0.9204	117	0.2252

Framework versions

Transformers 4.51.3
Pytorch 2.7.0+cu128
Datasets 3.5.1
Tokenizers 0.21.1