Spaces:
Sleeping
Sleeping
Update training/generate_plots.py
Browse files
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-
|
| 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 |
|