Add loss to progress bar output (official-stockfish#350)

vondele · web-flow · commit a7beb37b64e7 · 2025-09-24T06:34:59.000+02:00
diff --git a/model/lightning_module.py b/model/lightning_module.py
@@ -103,7 +103,7 @@ def step_(self, batch: tuple[Tensor, ...], batch_idx, loss_type):
             loss = loss * ((qf > pt) * p.qp_asymmetry + 1)
         loss = loss.mean()
 
-        self.log(loss_type, loss)
+        self.log(loss_type, loss, prog_bar=True)
 
         return loss
 
