lerobot/lerobot/common
Yachen Kang 294948eff5
fix losses unpadding bug
Use original_action_dim instead of self.config.max_action_dim when processing losses to exclude padded regions. Including padding areas in loss computation would impair the model's action reconstruction capability.
2025-04-10 15:29:32 +08:00
..
datasets [pre-commit.ci] pre-commit autoupdate (#871) 2025-04-08 06:58:46 +02:00
envs Support for PI0+FAST (#921) 2025-04-04 11:51:11 +02:00
optim chore(doc): add license header to all files (#818) 2025-03-05 17:56:51 +01:00
policies fix losses unpadding bug 2025-04-10 15:29:32 +08:00
robot_devices feat(visualization): replace cv2 GUI with Rerun (and solves ffmpeg versioning issues) (#903) 2025-04-09 17:33:01 +02:00
utils [pre-commit.ci] pre-commit autoupdate (#871) 2025-04-08 06:58:46 +02:00
constants.py chore(doc): add license header to all files (#818) 2025-03-05 17:56:51 +01:00