wangzixiang
/
lerobot
mirror of https://github.com/huggingface/lerobot.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Merge 2fe78bbfac into b43ece8934

This commit is contained in:
IrvingF7 2025-04-17 16:18:36 +02:00 committed by GitHub
parent b43ece8934 2fe78bbfac
commit d6fe46c5a0
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
lerobot/common/policies/pi0/modeling_pi0.py
View File

@ -330,8 +330,8 @@ class PI0Policy(PreTrainedPolicy):
# For backward pass # For backward pass
loss = losses.mean() loss = losses.mean()
# For logging # For logging. Use detach so won't create scalar to break graph when using torch.compile
loss_dict["l2_loss"] = loss.item() loss_dict["l2_loss"] = loss.detach()
return loss, loss_dict return loss, loss_dict