training_args.json

{
  "model_name": "meta-llama/Llama-3.1-8B",
  "stage1_method": "full",
  "stage2_method": "lora",
  "stage": "task_specific",
  "num_epochs": 3,
  "config_path": "/home/oelgendy/arabic-medical-llm/script/configs/lora.yaml",
  "stage1_checkpoint": "outputs/exp04_llama_full_lora/stage1",
  "train_samples": 16756,
  "val_samples": 882
}