Pikaliov/caption-test
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Remove eval_loss from train.csv (already in val.csv)

Browse Source 
Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
This commit is contained in:
pikaliov
2026-04-24 12:57:37 +03:00
parent d9c795d9ae
commit 536f904842
1 changed files with 0 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
src/training/train_gtauav.py
View File

@@ -755,7 +755,6 @@ def train(cfg: TrainConfigGTAUAV) -> None:
# Log train metrics to CSV (includes recall/AP if eval ran this epoch). # Log train metrics to CSV (includes recall/AP if eval ran this epoch).
train_row = {**means} train_row = {**means}
if train_recall: if train_recall:
train_row["eval_loss"] = train_recall.get("loss", 0.0)
train_row["r@1_q2g"] = train_recall.get("r@1_q2g", 0.0) train_row["r@1_q2g"] = train_recall.get("r@1_q2g", 0.0)
train_row["r@5_q2g"] = train_recall.get("r@5_q2g", 0.0) train_row["r@5_q2g"] = train_recall.get("r@5_q2g", 0.0)
train_row["r@10_q2g"] = train_recall.get("r@10_q2g", 0.0) train_row["r@10_q2g"] = train_recall.get("r@10_q2g", 0.0)