junaid0600 commited on
Commit
8ff0a54
·
verified ·
1 Parent(s): faf0863

Update training/generate_plots.py

Browse files
Files changed (1) hide show
  1. training/generate_plots.py +1 -1
training/generate_plots.py CHANGED
@@ -40,7 +40,7 @@ def plot_loss_curve(log_history: list, save_path: str = "loss_curve.png"):
40
  fig, (ax1, ax2) = plt.subplots(1, 2, figsize=(13, 5))
41
  fig.suptitle(
42
  "GRPO Training — SQL Database Engineer Agent\n"
43
- "Qwen2.5-1.5B fine-tuned with Unsloth + TRL",
44
  fontsize=13, fontweight="bold"
45
  )
46
 
 
40
  fig, (ax1, ax2) = plt.subplots(1, 2, figsize=(13, 5))
41
  fig.suptitle(
42
  "GRPO Training — SQL Database Engineer Agent\n"
43
+ "Qwen2.5-7B fine-tuned with Unsloth + TRL",
44
  fontsize=13, fontweight="bold"
45
  )
46