training_summary.json

{
  "project": "AEGIS AI",
  "dataset": "syed7741/aegis-industrial-ai-dataset",
  "base_model": "google/flan-t5-small",
  "fine_tuning_method": "LoRA / PEFT",
  "task_type": "SEQ_2_SEQ_LM",
  "device": "CPU",
  "original_records": 64,
  "training_records": 54,
  "evaluation_records": 10,
  "training_examples": 162,
  "evaluation_examples": 30,
  "epochs": 2,
  "batch_size": 1,
  "gradient_accumulation_steps": 4,
  "learning_rate": 0.0003,
  "lora_r": 4,
  "lora_alpha": 16,
  "lora_dropout": 0.05,
  "target_modules": [
    "q",
    "v"
  ],
  "history": [
    {
      "epoch": 1,
      "train_loss": 1.650111566355199,
      "eval_loss": 1.1663277645905812
    },
    {
      "epoch": 2,
      "train_loss": 1.18804861218841,
      "eval_loss": 0.77438918252786
    }
  ]
}