admin/lerobot
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-07-24 10:16:09 +00:00
Code Issues Packages Projects Releases Wiki Activity

rename to loss

Browse Source
This commit is contained in:
Pepijn
2025-09-13 16:15:29 +02:00
parent 7d9b469eee
commit d0d714be47
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/lerobot/policies/pi0_openpi/modeling_pi0openpi.py
+1 -1
View File
@@ -1314,7 +1314,7 @@ class PI0OpenPIPolicy(PreTrainedPolicy):
loss = losses.mean() loss = losses.mean()
loss_dict = { loss_dict = {
"l2_loss": loss.item(), "loss": loss.item(),
"loss_per_dim": losses.mean(dim=[0, 1]).detach().cpu().numpy().tolist(), "loss_per_dim": losses.mean(dim=[0, 1]).detach().cpu().numpy().tolist(),
} }