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

Rename 'total' to 'train_loss' in train.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:59:47 +03:00
parent 536f904842
commit e0db8d2726
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/training/train_gtauav.py
View File

@@ -754,6 +754,8 @@ 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 "total" in train_row:
train_row["train_loss"] = train_row.pop("total")
if train_recall: if train_recall:
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)