From d0d714be4784870bee97c8226f089670a0323b21 Mon Sep 17 00:00:00 2001 From: Pepijn Date: Sat, 13 Sep 2025 16:15:29 +0200 Subject: [PATCH] rename to loss --- src/lerobot/policies/pi0_openpi/modeling_pi0openpi.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/lerobot/policies/pi0_openpi/modeling_pi0openpi.py b/src/lerobot/policies/pi0_openpi/modeling_pi0openpi.py index 4bb1c83d9..8cfa1743b 100644 --- a/src/lerobot/policies/pi0_openpi/modeling_pi0openpi.py +++ b/src/lerobot/policies/pi0_openpi/modeling_pi0openpi.py @@ -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(), }