mirror of
https://github.com/huggingface/lerobot.git
synced 2026-06-30 06:37:15 +00:00
fix(rl): update neutral gripper action
(cherry picked from commit 9c9064e5be)
This commit is contained in:
@@ -691,7 +691,7 @@ def control_loop(
|
||||
# Create a neutral action (no movement)
|
||||
neutral_action = torch.tensor([0.0, 0.0, 0.0], dtype=torch.float32)
|
||||
if use_gripper:
|
||||
neutral_action = torch.cat([neutral_action, torch.tensor([0.0])]) # Gripper stay
|
||||
neutral_action = torch.cat([neutral_action, torch.tensor([1.0])]) # Gripper stay
|
||||
|
||||
transition = step_env_and_process_transition(
|
||||
env=env,
|
||||
|
||||
Reference in New Issue
Block a user