cyy/go2_rl_robotlab
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix bugs in actions; update readme.

Browse Source
This commit is contained in:
wertyuilife
2026-03-31 00:19:10 +08:00
parent 89dce4524c
commit fea45ce4bd
3 changed files with 7 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
source/robot_lab/robot_lab/tasks/go2/env/go2_env.py vendored
View File

@@ -64,7 +64,7 @@ class ActionDelayGo2Env(ManagerBasedRLEnv):
self._sim_step_counter += 1
# determine which envs should apply delayed action at this decimation step
action_delay_masks = (i <= actions_start_decimation)
action_delay_masks = (i < actions_start_decimation)
self.action_manager.process_action_with_delay(action_delay_masks)
# set actions into buffers

5
source/robot_lab/robot_lab/tasks/go2/manager/action_manager.py
View File

@@ -9,7 +9,10 @@ class ActionManagerWithDelay(ActionManager):
def reset(self, env_ids: Sequence[int] | None = None) -> dict[str, torch.Tensor]:
super().reset(env_ids)
self._prev_prev_action.zero_()
if env_ids is None:
self._prev_prev_action.zero_()
else:
self._prev_prev_action[env_ids] = 0.0
return {}
def process_action(self, action: torch.Tensor):