Added option to add current readings to the state of the policy

This commit is contained in:
Michel Aractingi
2025-04-15 15:07:43 +02:00
parent 3b24ad3c84
commit 9886520d33
3 changed files with 69 additions and 16 deletions

View File

@@ -192,6 +192,7 @@ class EnvWrapperConfig:
display_cameras: bool = False
use_relative_joint_positions: bool = True
add_joint_velocity_to_observation: bool = False
add_current_to_observation: bool = False
add_ee_pose_to_observation: bool = False
crop_params_dict: Optional[Dict[str, Tuple[int, int, int, int]]] = None
resize_size: Optional[Tuple[int, int]] = None