From 7bb0aeb756bfb7485357ade17e66cdf980e476da Mon Sep 17 00:00:00 2001 From: Roland Tannous Date: Tue, 3 Feb 2026 10:35:58 +0000 Subject: [PATCH] add grad_norm and num_tokens to TrainingProgress response object --- studio/backend/models/training.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/studio/backend/models/training.py b/studio/backend/models/training.py index 86aead6462..2afee72e89 100644 --- a/studio/backend/models/training.py +++ b/studio/backend/models/training.py @@ -97,4 +97,6 @@ class TrainingProgress(BaseModel): epoch: Optional[int] = Field(None, description="Current epoch") elapsed_seconds: Optional[float] = Field(None, description="Time elapsed since training started") eta_seconds: Optional[float] = Field(None, description="Estimated time remaining") + grad_norm: Optional[float] = Field(None, description="L2 norm of gradients, computed before gradient clipping") + num_tokens: Optional[int] = Field(None, description="Total number of tokens processed so far")