ydy0615/lerobot-clone
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-06-03 20:31:25 +00:00
Code Issues Packages Projects Releases Wiki Activity

gixes for gripper penalty

Browse Source
This commit is contained in:
Michel Aractingi
2025-04-07 16:51:17 +02:00
parent 90a30ed319
commit 06de182448
5 changed files with 12 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
lerobot/scripts/server/learner_server.py
View File

@@ -408,6 +408,7 @@ def add_actor_information_and_train(
"done": done,
"observation_feature": observation_features,
"next_observation_feature": next_observation_features,
"complementary_info": batch.get("complementary_info", None),
}
# Use the forward method for critic loss (includes both main critic and grasp critic)