From e0db8d272640c99bb868abc9aba903ed6f895d1a Mon Sep 17 00:00:00 2001 From: pikaliov Date: Fri, 24 Apr 2026 12:59:47 +0300 Subject: [PATCH] Rename 'total' to 'train_loss' in train.csv Co-Authored-By: Claude Opus 4.6 (1M context) --- src/training/train_gtauav.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/src/training/train_gtauav.py b/src/training/train_gtauav.py index b12e614..492dc91 100644 --- a/src/training/train_gtauav.py +++ b/src/training/train_gtauav.py @@ -754,6 +754,8 @@ def train(cfg: TrainConfigGTAUAV) -> None: # Log train metrics to CSV (includes recall/AP if eval ran this epoch). train_row = {**means} + if "total" in train_row: + train_row["train_loss"] = train_row.pop("total") if train_recall: 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)