rename to loss

2026-07-05 17:17:01 +00:00 · 2025-09-13 16:15:29 +02:00
parent 7d9b469eee
commit d0d714be47
1 changed files with 1 additions and 1 deletions
@@ -1314,7 +1314,7 @@ class PI0OpenPIPolicy(PreTrainedPolicy):
        loss = losses.mean()

        loss_dict = {
-            "l2_loss": loss.item(),
+            "loss": loss.item(),
            "loss_per_dim": losses.mean(dim=[0, 1]).detach().cpu().numpy().tolist(),
        }