lerobot/lerobot/scripts/server
Michel Aractingi a7db3959f5 - Added JointMaskingActionSpace wrapper in `gym_manipulator` in order to select which joints will be controlled. For example, we can disable the gripper actions for some tasks.
- Added Nan detection mechanisms in the actor, learner and gym_manipulator for the case where we encounter nans in the loop.
- changed the non-blocking in the `.to(device)` functions to only work for the case of cuda because they were causing nans when running the policy on mps
- Added some joint clipping and limits in the env, robot and policy configs. TODO clean this part and make the limits in one config file only.

Co-authored-by: Adil Zouitine <adilzouitinegm@gmail.com>
2025-02-11 11:34:46 +01:00
..
wrappers Added additional wrappers for the environment: Action repeat, keyboard interface, reset wrapper 2025-02-04 17:41:14 +00:00
actor_server.py - Added JointMaskingActionSpace wrapper in `gym_manipulator` in order to select which joints will be controlled. For example, we can disable the gripper actions for some tasks. 2025-02-11 11:34:46 +01:00
buffer.py - Added JointMaskingActionSpace wrapper in `gym_manipulator` in order to select which joints will be controlled. For example, we can disable the gripper actions for some tasks. 2025-02-11 11:34:46 +01:00
crop_dataset_roi.py fixed bug in crop_dataset_roi.py 2025-02-05 18:22:50 +00:00
find_joint_limits.py - Added JointMaskingActionSpace wrapper in `gym_manipulator` in order to select which joints will be controlled. For example, we can disable the gripper actions for some tasks. 2025-02-11 11:34:46 +01:00
gym_manipulator.py - Added JointMaskingActionSpace wrapper in `gym_manipulator` in order to select which joints will be controlled. For example, we can disable the gripper actions for some tasks. 2025-02-11 11:34:46 +01:00
hilserl.proto - Added additional logging information in wandb around the timings of the policy loop and optimization loop. 2025-02-03 15:07:58 +00:00
learner_server.py - Added JointMaskingActionSpace wrapper in `gym_manipulator` in order to select which joints will be controlled. For example, we can disable the gripper actions for some tasks. 2025-02-11 11:34:46 +01:00