Skip to content
Open
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
5 changes: 3 additions & 2 deletions humanoidverse/envs/motion_tracking/motion_tracking.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,7 +5,7 @@
from humanoidverse.envs.legged_base_task.legged_robot_base import LeggedRobotBase
from isaac_utils.rotations import (
my_quat_rotate,
quat_to_tan_norm,
# quat_to_tan_norm,
calc_heading_quat_inv,
calc_heading_quat,
quat_mul,
Expand Down Expand Up @@ -610,7 +610,8 @@ def _reward_teleop_body_position_feet(self):
return r_feet

def _reward_teleop_body_rotation_extend(self):
rotation_diff = quat_to_angle_axis(self.dif_global_body_rot, w_last=True)[0]
# rotation_diff = quat_to_angle_axis(self.dif_global_body_rot, w_last=True)[0]
rotation_diff = quat_to_angle_axis(self.dif_global_body_rot)[0]
diff_body_rot_dist = (rotation_diff**2).mean(dim=-1)
r_body_rot = torch.exp(-diff_body_rot_dist / self.config.rewards.reward_tracking_sigma.teleop_body_rot)
return r_body_rot
Expand Down