{
  "base_model": "meta-llama/Llama-3.2-1B",
  "hidden_size": 2048,
  "ffn_hidden": 4096,
  "lora_r": 16,
  "lora_alpha": 32,
  "lora_dropout": 0.05,
  "lora_targets": [
    "q_proj",
    "v_proj",
    "k_proj",
    "o_proj"
  ],
  "num_experts": 4,
  "top_k": 2,
  "dataset_name": "Malikeh1375/medical-question-answering-datasets",
  "dataset_subset": "all-processed",
  "num_samples": 15000,
  "max_length": 256,
  "batch_size": 2,
  "grad_accum": 16,
  "warmup_ratio": 0.1,
  "val_split": 0.1,
  "save_path": "/kaggle/working/llama_moe_medical",
  "model_type": "llama-3.2-1b-lora-moe",
  "lora_merged": true,
  "hf_username": "saadxsalman",
  "repo_id": "saadxsalman/ss-medical-moe",
  "domain_experts": {
    "expert_1": "Symptoms & Diagnosis",
    "expert_2": "Drug Interactions",
    "expert_3": "Anatomy & Physiology",
    "expert_4": "Treatment & Procedures"
  }
}