wangzixiang
/
lerobot
mirror of https://github.com/huggingface/lerobot.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
919 Commits 131 Branches 0 Tags 244 MiB
Commit Graph

637 Commits

Author SHA1 Message Date
Michel Aractingi 508f5d1407 Added server directory in `lerobot/scripts` that contains scripts and the protobuf message types to split training into two processes, acting and learning. The actor rollouts the policy and collects interaction data while the learner recieves the data, trains the policy and sends the updated parameters to the actor. The two scripts are ran simultaneously 
Co-authored-by: Adil Zouitine <adilzouitinegm@gmail.com>
 2025-04-18 15:04:13 +02:00
AdilZouitine c8b1132846 Stable version of rlpd + drq 2025-04-18 15:04:10 +02:00
AdilZouitine ef777993cd Add type annotations and restructure SACConfig class fields 2025-04-18 15:03:51 +02:00
Adil Zouitine 760d60ad4b Change SAC policy implementation with configuration and modeling classes 2025-04-18 15:03:51 +02:00
Adil Zouitine 875c0271b7 SAC works 2025-04-18 15:03:51 +02:00
Adil Zouitine 57344bfde5 [WIP] correct sac implementation 2025-04-18 15:03:51 +02:00
Adil Zouitine 46827fb002 Add rlpd tricks 2025-04-18 15:03:51 +02:00
Adil Zouitine 2fd78879f6 SAC works 2025-04-18 15:03:51 +02:00
Adil Zouitine e8449e9630 remove breakpoint 2025-04-18 15:03:51 +02:00
Adil Zouitine a0e2be8b92 [WIP] correct sac implementation 2025-04-18 15:03:51 +02:00
Michel Aractingi 181727c0fe Extend reward classifier for multiple camera views (#626) 2025-04-18 15:03:50 +02:00
Eugene Mironov d1d6ffd23c [Port HIL_SERL] Final fixes for the Reward Classifier (#598) 2025-04-18 15:03:01 +02:00
Michel Aractingi e5801f467f added temporary fix for missing task_index key in online environment 2025-04-18 15:03:01 +02:00
Michel Aractingi c6ca9523de split encoder for critic and actor 2025-04-18 15:03:01 +02:00
Michel Aractingi 642e3a3274 style fixes 2025-04-18 15:03:01 +02:00
KeWang1017 146148c48c Refactor SAC configuration and policy for improved action sampling and stability 
- Updated SACConfig to replace standard deviation parameterization with log_std_min and log_std_max for better control over action distributions.
- Modified SACPolicy to streamline action selection and log probability calculations, enhancing stochastic behavior.
- Removed deprecated TanhMultivariateNormalDiag class to simplify the codebase and improve maintainability.

These changes aim to enhance the robustness and performance of the SAC implementation during training and inference.
 2025-04-18 15:03:01 +02:00
KeWang1017 8f15835daa Refine SAC configuration and policy for enhanced performance 
- Updated standard deviation parameterization in SACConfig to 'softplus' with defined min and max values for improved stability.
- Modified action sampling in SACPolicy to use reparameterized sampling, ensuring better gradient flow and log probability calculations.
- Cleaned up log probability calculations in TanhMultivariateNormalDiag for clarity and efficiency.
- Increased evaluation frequency in YAML configuration to 50000 for more efficient training cycles.

These changes aim to enhance the robustness and performance of the SAC implementation during training and inference.
 2025-04-18 15:03:01 +02:00
KeWang1017 022bd65125 Refactor SACPolicy for improved action sampling and standard deviation handling 
- Updated action selection to use distribution sampling and log probabilities for better stochastic behavior.
- Enhanced standard deviation clamping to prevent extreme values, ensuring stability in policy outputs.
- Cleaned up code by removing unnecessary comments and improving readability.

These changes aim to refine the SAC implementation, enhancing its robustness and performance during training and inference.
 2025-04-18 15:03:01 +02:00
KeWang1017 63d8c96514 trying to get sac running 2025-04-18 15:03:01 +02:00
Michel Aractingi 4624a836e5 Added normalization schemes and style checks 2025-04-18 15:03:01 +02:00
Michel Aractingi ad7eea132d added optimizer and sac to factory.py 2025-04-18 15:02:59 +02:00
Eugene Mironov 22a1899ff4 [HIL-SERL PORT] Fix linter issues (#588) 2025-04-18 15:02:44 +02:00
Eugene Mironov 17a3a31b5f [Port Hil-SERL] Add unit tests for the reward classifier & fix imports & check script (#578) 2025-04-18 15:02:42 +02:00
Michel Aractingi 1a8b99e360 added comments from kewang 2025-04-18 15:02:13 +02:00
KeWang1017 6db2154f28 Enhance SAC configuration and policy with new parameters and subsampling logic 
- Added `num_subsample_critics`, `critic_target_update_weight`, and `utd_ratio` to SACConfig.
- Implemented target entropy calculation in SACPolicy if not provided.
- Introduced subsampling of critics to prevent overfitting during updates.
- Updated temperature loss calculation to use the new target entropy.
- Added comments for future UTD update implementation.

These changes improve the flexibility and performance of the SAC implementation.
 2025-04-18 15:02:13 +02:00
KeWang be3adda95f Port SAC WIP (#581) 
Co-authored-by: KeWang1017 <ke.wang@helloleap.ai>
 2025-04-18 15:02:13 +02:00
Michel Aractingi 9d48d236c1 completed losses 2025-04-18 15:02:13 +02:00
Michel Aractingi b57d6a7776 nit in control_robot.py 2025-04-18 15:02:13 +02:00
Michel Aractingi d1f76cba8e Update lerobot/scripts/train_hilserl_classifier.py 
Co-authored-by: Yoel <yoel.chornton@gmail.com>
 2025-04-18 15:02:13 +02:00
Eugene Mironov d78cef1fee Fixup 2025-04-18 15:02:13 +02:00
Michel Aractingi 30a808c0ae Add human intervention mechanism and eval_robot script to evaluate policy on the robot (#541) 
Co-authored-by: Yoel <yoel.chornton@gmail.com>
 2025-04-18 15:02:13 +02:00
Yoel 4a7f85a6ec Reward classifier and training (#528) 
Co-authored-by: Daniel Ritchie <daniel@brainwavecollective.ai>
Co-authored-by: resolver101757 <kelster101757@hotmail.com>
Co-authored-by: Jannik Grothusen <56967823+J4nn1K@users.noreply.github.com>
Co-authored-by: Remi <re.cadene@gmail.com>
Co-authored-by: Michel Aractingi <michel.aractingi@huggingface.co>
 2025-04-18 15:02:13 +02:00
Alex Thiele c10c5a0e64
Fix --width --height type parsing on opencv and intelrealsense scripts (#556) 
Co-authored-by: Remi <remi.cadene@huggingface.co>
Co-authored-by: Steven Palma <imstevenpmwork@ieee.org>
 2025-04-17 15:19:23 +02:00
Junshan Huang a8db91c40e
Fix Windows HTML visualization to make videos could be seen (#647) 
Co-authored-by: Simon Alibert <75076266+aliberts@users.noreply.github.com>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Steven Palma <imstevenpmwork@ieee.org>
 2025-04-17 15:07:28 +02:00
Steven Palma 4041f57943
feat(visualization): replace cv2 GUI with Rerun (and solves ffmpeg versioning issues) (#903) 2025-04-09 17:33:01 +02:00
pre-commit-ci[bot] 437fc29e12
[pre-commit.ci] pre-commit autoupdate (#871) 
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
 2025-04-08 06:58:46 +02:00
mshukor 1c873df5c0
Support for PI0+FAST (#921) 
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Dana Aubakirova <118912928+danaaubakirova@users.noreply.github.com>
Co-authored-by: Remi <re.cadene@gmail.com>
Co-authored-by: Steven Palma <imstevenpmwork@ieee.org>
 2025-04-04 11:51:11 +02:00
Steven Palma b568de35ad
fix(datasets): cast imgs_dir as Path (#915) 2025-03-28 18:08:12 +01:00
Steven Palma 78fd1a1e04
chore(docs): update docs (#911) 2025-03-27 09:55:06 +01:00
AlexC 2c22f7d76d
Add offline mode in the configuration for wandb logging (#897) 
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Steven Palma <imstevenpmwork@ieee.org>
 2025-03-25 13:44:49 +01:00
Qizhi Chen a774af2eab
fix pi0 action padding name (#893) 
Co-authored-by: Steven Palma <imstevenpmwork@ieee.org>
 2025-03-25 11:24:46 +01:00
Steven Palma a6015a55f9
chore(scripts): remove deprecated script (#887) 2025-03-23 01:16:50 +01:00
Pepijn 952e892fe5
Use float32 instead of int (#877) 2025-03-18 16:36:37 +01:00
Pepijn e8159997c7
User/pepijn/2025 03 17 act different image shapes (#870) 2025-03-18 11:09:05 +01:00
Steven Palma 1c15bab70f
fix(codec): hot-fix for default codec in linux arm platforms (#868) 2025-03-17 13:23:11 +01:00
Huan Liu a3cd18eda9
added wandb.run_id to allow resuming without wandb log; updated log m… (#841) 
Co-authored-by: Simon Alibert <75076266+aliberts@users.noreply.github.com>
 2025-03-15 09:40:39 +01:00
Jade Choghari 0e98c6ee96
Add torchcodec cpu (#798) 
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Remi <re.cadene@gmail.com>
Co-authored-by: Remi <remi.cadene@huggingface.co>
Co-authored-by: Simon Alibert <simon.alibert@huggingface.co>
Co-authored-by: Simon Alibert <75076266+aliberts@users.noreply.github.com>
 2025-03-14 16:53:42 +01:00
Simon Alibert 974028bd28
Organize test folders (#856) 
Co-authored-by: Steven Palma <imstevenpmwork@ieee.org>
 2025-03-13 14:05:55 +01:00
Ermano Arruda c37b1d45b6
parametrise tolerance_s in visualize_dataset scripts (#716) 2025-03-13 10:28:29 +01:00
Pepijn 84565c7c2e
Fix camera rotation error (#839) 
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
 2025-03-10 17:02:19 +01:00
Ben Sprenger 05b54733da
feat: add support for external plugin config dataclasses (#807) 
Co-authored-by: Steven Palma <imstevenpmwork@ieee.org>
Co-authored-by: Simon Alibert <75076266+aliberts@users.noreply.github.com>
 2025-03-10 13:25:47 +01:00
Joe Clinton 32fffd4bbb
Fix delay in teleoperation start time (#676) 
Co-authored-by: Simon Alibert <75076266+aliberts@users.noreply.github.com>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
 2025-03-08 11:40:07 +01:00
Steven Palma 5e9473806c
refactor(config): Move device & amp args to PreTrainedConfig (#812) 
Co-authored-by: Simon Alibert <75076266+aliberts@users.noreply.github.com>
 2025-03-06 17:59:28 +01:00
Harsimrat Sandhawalia 10706ed753
Support for discrete actions (#810) 2025-03-06 10:27:29 +01:00
Steven Palma 5d24ce3160
chore(doc): add license header to all files (#818) 2025-03-05 17:56:51 +01:00
yadunund 2feb5edc65
Fix printout in make_cameras_from_configs (#796) 
Signed-off-by: Yadunund <yadunund@gmail.com>
Co-authored-by: Steven Palma <imstevenpmwork@ieee.org>
 2025-03-05 10:01:24 +01:00
Yachen Kang b80e55ca44
change "actions_id_pad" to "actions_is_pad"(🐛 Bug) (#774) 
Co-authored-by: Steven Palma <imstevenpmwork@ieee.org>
 2025-03-05 01:31:56 +01:00
Pepijn a003e7c081
change wheel setup in kinematics (#811) 
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
 2025-03-04 18:42:45 +01:00
Mishig a27411022d
[visualization] Ignore 2d or 3d data for now (#809) 2025-03-04 10:53:01 +01:00
Simon Alibert 8861546ad8
[Security] Add Bandit (#795) 2025-03-01 19:19:26 +01:00
Simon Alibert e81c36cf74
Fix dataset version tags (#790) 2025-02-28 14:36:20 +01:00
Mishig 800c4a847f
[Vizualisation] independent column names (#783) 2025-02-27 14:47:18 +01:00
Mishig 8d60ac3ffc
[vizualisation] Add pagination for many episodes (#776) 2025-02-26 19:23:37 +01:00
Simon Alibert a1809ad3de
Add typos checks (#770) 2025-02-25 23:51:15 +01:00
Jannik Grothusen 8699a28be0
[QOL] Enable teleoperation during environment reset (#725) 2025-02-25 19:28:26 +01:00
Haskely eecf32e77a
feat: Add root directory option for dataset configuration (#765) 
Co-authored-by: Simon Alibert <75076266+aliberts@users.noreply.github.com>
 2025-02-25 17:27:36 +01:00
Simon Alibert 3354d919fc
LeRobotDataset v2.1 (#711) 
Co-authored-by: Remi <remi.cadene@huggingface.co>
Co-authored-by: Remi Cadene <re.cadene@gmail.com>
 2025-02-25 15:27:29 +01:00
Pepijn aca464ca72
Add mobile so100 (#724) 2025-02-25 09:06:50 +01:00
Simon Alibert c4c2ce04e7
Update pre-commits (#733) 2025-02-15 15:51:17 +01:00
Simon Alibert b86a2c0b47
Fix wandb logging (#730) 2025-02-14 18:00:12 +01:00
Ilia Larchenko c574eb4984
Fixed eval.py on MPS (#702) 2025-02-14 00:03:55 +01:00
Simon Alibert 1e49cc4d60
Prevent resuming from hub (#726) 2025-02-13 17:15:55 +01:00
Simon Alibert e71095960f
Fixes following #670 (#719) 2025-02-12 12:53:55 +01:00
Simon Alibert 90e099b39f
Remove offline training, refactor `train.py` and logging/checkpointing (#670) 
Co-authored-by: Remi <remi.cadene@huggingface.co>
 2025-02-11 10:36:06 +01:00
Remi 638d411cd3
Add Pi0 (#681) 
Co-authored-by: Simon Alibert <simon.alibert@huggingface.co>
Co-authored-by: Simon Alibert <75076266+aliberts@users.noreply.github.com>
Co-authored-by: Pablo <pablo.montalvo.leroux@gmail.com>
 2025-02-04 18:01:04 +01:00
Pepijn dd974529cf
User/pepijn/2025 01 31 improved tutorial so100 (#666) 2025-02-03 18:27:55 +01:00
Pepijn ae9605f03c
fix setting motor id with new dataclass config (#668) 2025-01-31 20:48:46 +01:00
Simon Alibert 3c0a209f9f
Simplify configs (#550) 
Co-authored-by: Remi <remi.cadene@huggingface.co>
Co-authored-by: HUANG TZU-CHUN <137322177+tc-huang@users.noreply.github.com>
 2025-01-31 13:57:37 +01:00
Simon Alibert 1ee1acf8ad
Comply with torchvision 0.21 custom transforms (#665) 2025-01-30 22:06:11 +01:00
Thomas Lips c4d912a241
Check for "/" in feature names (#660) 2025-01-29 21:54:49 +01:00
Morgan Redfield 4323bdce22
updating config instructions for koch 1v1 motors (#658) 2025-01-28 13:20:33 +01:00
HUANG TZU-CHUN 5daa45436d
Fix typos in `lerobot/scripts/visualize_dataset.py` (#656) 2025-01-28 13:07:10 +01:00
Pradeep Kadubandi 380b836eee
Fix for the issue https://github.com/huggingface/lerobot/issues/638 (#639) 2025-01-15 10:50:38 +01:00
Mishig 25a8597680
[viz] Fixes & updates to html visualizer (#617) 2025-01-09 11:39:54 +01:00
CharlesCNorton b8b368310c
typo fix: batch_convert_dataset_v1_to_v2.py (#615) 
Co-authored-by: Simon Alibert <75076266+aliberts@users.noreply.github.com>
 2025-01-09 09:57:45 +01:00
Ville Kuosmanen 5097cd900e
fix(visualise): use correct language description for each episode id (#604) 
Co-authored-by: Simon Alibert <75076266+aliberts@users.noreply.github.com>
 2025-01-09 09:39:48 +01:00
Eugene Mironov 59e2757434
Fix broken `create_lerobot_dataset_card` (#590) 2024-12-23 15:05:59 +01:00
Mishig 73b64c3089
[vizualizer] for LeRobodDataset V2 (#576) 2024-12-20 16:26:23 +01:00
s1lent4gnt 66f8736598
fixing typo from 'teloperation' to 'teleoperation' (#566) 2024-12-11 05:57:52 -08:00
Simon Alibert 4c41f6fcc6
Fix example 6 (#572) 2024-12-11 10:32:18 +01:00
Claudio Coppola 44f9b21e74
LerobotDataset pushable to HF from any folder (#563) 2024-12-09 11:32:25 +01:00
Michel Aractingi 8e7d6970ea
Control simulated robot with real leader (#514) 
Co-authored-by: Remi <remi.cadene@huggingface.co>
 2024-12-03 12:20:05 +01:00
Remi 286bca37cc
Fix missing local_files_only in record/replay (#540) 
Co-authored-by: Simon Alibert <alibert.sim@gmail.com>
 2024-12-03 10:53:21 +01:00
Michel Aractingi a2c181992a
Refactor OpenX (#505) 2024-12-03 00:51:55 +01:00
Simon Alibert 32eb0cec8f
Dataset v2.0 (#461) 
Co-authored-by: Remi <remi.cadene@huggingface.co>
 2024-11-29 19:04:00 +01:00
KasparSLT 96c7052777
Rename deprecated argument (temporal_ensemble_momentum) (#490) 2024-11-25 21:05:13 +01:00
Jannik Grothusen 975c1c25c3
Add distinction between two unallowed cases in name check "eval_" (#489) 2024-11-22 19:19:57 +01:00
Daniel Ritchie 8af693548e
Add support for Windows (#494) 2024-11-22 19:14:25 +01:00
Hirokazu Ishida 538455a965
feat: enable to use multiple rgb encoders per camera in diffusion policy (#484) 
Co-authored-by: Alexander Soare <alexander.soare159@gmail.com>
 2024-10-30 11:00:05 +01:00
Remi 172809a502
[Fix] Move back to manual calibration (#488) 2024-10-26 15:27:21 +02:00
Remi 55e4ff6742
Fix autocalib moss (#486) 2024-10-26 12:15:17 +02:00
Remi 07e8716315
Add FeetechMotorsBus, SO-100, Moss-v1 (#419) 
Co-authored-by: jess-moss <jess.moss@huggingface.co>
Co-authored-by: Simon Alibert <75076266+aliberts@users.noreply.github.com>
 2024-10-25 11:23:55 +02:00
Alexander Soare cd0fc261c0
Make `say(blocking=True)` work for Linux (#460) 2024-10-17 15:22:21 +01:00
Remi 77478d50e5
Refactor `record` with `add_frame` (#468) 
Co-authored-by: Simon Alibert <75076266+aliberts@users.noreply.github.com>
 2024-10-16 20:51:35 +02:00
Remi 97b1feb0b3
Add policy/act_aloha_real.yaml + env/act_real.yaml (#429) 
Co-authored-by: Simon Alibert <75076266+aliberts@users.noreply.github.com>
 2024-10-10 17:12:45 +02:00
Eugene Mironov c29e70e5a1
Fix issue with wrong using index instead of camera_index in opencv (#466) 2024-10-09 11:35:19 +02:00
Simon Alibert 1a343c3591
Add support for Stretch (hello-robot) (#409) 
Co-authored-by: Remi <remi.cadene@huggingface.co>
Co-authored-by: Remi Cadene <re.cadene@gmail.com>
 2024-10-04 18:56:42 +02:00
Remi 26f97cfd17
Enable CI for robot devices with mocked versions (#398) 
Co-authored-by: Simon Alibert <75076266+aliberts@users.noreply.github.com>
 2024-10-03 17:05:23 +02:00
Simon Alibert 72f402d44b
Fix dataset card (#453) 2024-09-25 16:56:05 +02:00
Alexander Soare 92573486a8
Don't use async envs by default (#448) 2024-09-20 15:22:52 +02:00
Simon Alibert c712d68f6a
Fix nightlies (#443) 2024-09-18 14:51:45 +02:00
Dana Aubakirova f431a08efa
small fix: assertion error message in envs/utils.py (#426) 
Co-authored-by: Remi <re.cadene@gmail.com>
Co-authored-by: Alexander Soare <alexander.soare159@gmail.com>
Co-authored-by: Remi <remi.cadene@huggingface.co>
 2024-09-12 18:03:34 +02:00
Remi beaa427504
Fix slow camera fps with Aloha (#433) 2024-09-12 14:20:24 +02:00
Mishig a88dd602d9
[Vizualization] Better error message (#430) 
Co-authored-by: Simon Alibert <75076266+aliberts@users.noreply.github.com>
 2024-09-12 10:46:48 +02:00
Mishig 6c0324f467
[Vizualization] Fix video layout (#431) 2024-09-12 10:06:29 +02:00
Alexander Soare a60d27b132
Raise ValueError if horizon is incompatible with downsampling (#422) 2024-09-09 17:22:46 +01:00
Mishig 9c463661c1
[Vizualization] Better UI on small screens (like in smartphones) (#423) 2024-09-09 15:39:40 +02:00
Mishig 4255655618
[Vizualization] Show user error if videos codec is not supported (#424) 2024-09-09 15:38:41 +02:00
Joe Clinton f17d9a2ba1
Bug: Fix VQ-Bet not working when n_action_pred_token=1 (#420) 
Co-authored-by: Alexander Soare <alexander.soare159@gmail.com>
 2024-09-09 09:41:13 +01:00
Remi 9ff829a3a1
Add comments for Aloha (#417) 
Co-authored-by: Simon Alibert <75076266+aliberts@users.noreply.github.com>
 2024-09-06 21:07:52 +02:00
Mishig d6516f0e03
[Visualization tool] Fix when dim state != dim action (#415) 2024-09-06 17:07:26 +02:00
Jack Vial b0b8612eff
fix(calibrate): fix calibrate arms option type. should be str not int (#418) 
Co-authored-by: Remi <remi.cadene@huggingface.co>
 2024-09-06 14:44:31 +02:00
Mishig 1072a055db
[Visualization tool] Fix videos sync (#416) 2024-09-06 10:16:08 +02:00
Remi 9c9f5cac90
Add IntelRealSenseCamera (#410) 
Co-authored-by: Simon Alibert <simon.alibert@huggingface.co>
Co-authored-by: shantanuparab-tr <shantanu@trossenrobotics.com>
Co-authored-by: Simon Alibert <75076266+aliberts@users.noreply.github.com>
 2024-09-05 23:59:41 +02:00
Simon Alibert 9d0c6fe419
Fix nightlies & untrack json files from git lfs (#414) 2024-09-05 15:07:43 +02:00
Remi 429a463aff
Control aloha robot natively (#316) 
Co-authored-by: Simon Alibert <75076266+aliberts@users.noreply.github.com>
 2024-09-04 19:28:05 +02:00
Jack Vial 27ba2951d1
fix(tdmpc): Add missing save_freq to tdmpc policy config (#404) 
Co-authored-by: Alexander Soare <alexander.soare159@gmail.com>
 2024-09-02 19:04:41 +01:00
Jack Vial b2896d38f5
fix(act): n_vae_encoder_layers config parameter wasn't being used (#400) 2024-09-02 18:29:27 +01:00
Mishig 114e09f570
rm EpisodeSampler from viz (#389) 2024-08-30 10:53:55 +02:00
Simon Alibert 04a995e7d1
Fix safe_action (#395) 2024-08-30 10:36:05 +02:00
Michel Aractingi 4806336816
Add the possibility to visualize language instructions in visualize_dataset_html.py (#388) 
Co-authored-by: Mishig <dmishig@gmail.com>
 2024-08-28 11:50:31 +02:00
Remi 1ce418e4a1
Add koch bimanual (#385) 2024-08-28 00:53:31 +02:00
Michel Aractingi eb4c505cff
Support for converting OpenX datasets from RLDS format to LeRobotDataset (#354) 
Signed-off-by: youliangtan <tan_you_liang@hotmail.com>
Co-authored-by: Simon Alibert <alibert.sim@gmail.com>
Co-authored-by: youliangtan <tan_you_liang@hotmail.com>
Co-authored-by: Remi <re.cadene@gmail.com>
 2024-08-27 09:07:00 +02:00
Mishig aad59e6b6b
Fix videos in visualize_dataset are not in sync (#382) 2024-08-26 17:38:48 +02:00
Alexander Soare 9ce98bb93c
Add safety limits on relative action target (#373) 2024-08-26 14:30:18 +01:00
Alexander Soare 97086cdcdf
Make gripper_open_degree a config param (#379) 2024-08-26 12:28:16 +01:00
Alexander Soare 9c7649f140
Make sure `init_hydra_config` does not require any keys (#376) 2024-08-23 12:27:08 +01:00
Zhuoheng Li a2592a5563
Provide more information to the user (#358) 
Co-authored-by: Alexander Soare <alexander.soare159@gmail.com>
Co-authored-by: Remi <re.cadene@gmail.com>
 2024-08-23 11:00:35 +01:00
NielsRogge 86bbd16d43
Improve discoverability on the hub (#325) 
Co-authored-by: Lucain <lucainp@gmail.com>
Co-authored-by: Simon Alibert <75076266+aliberts@users.noreply.github.com>
 2024-08-19 15:16:46 +02:00
Alexander Soare 0f6e0f6d74
Fix input dim (#365) 2024-08-19 11:42:32 +01:00
Simon Alibert b98ea415c1
Add dataset cards (#363) 2024-08-16 10:08:44 +02:00
Remi bbe9057225
Improve control robot ; Add process to configure motor indices (#326) 
Co-authored-by: Simon Alibert <alibert.sim@gmail.com>
Co-authored-by: jess-moss <jess.moss@dextrousrobotics.com>
Co-authored-by: Marina Barannikov <marina.barannikov@huggingface.co>
Co-authored-by: Alexander Soare <alexander.soare159@gmail.com>
 2024-08-15 18:11:33 +02:00
Remi 2252b42337
Add visualize_dataset_html with `http.server` (#188) 2024-08-08 20:19:06 +03:00
Halvard Bariller 7a3cb1ad34
Adjust the timestamps' description in Diffusion Policy (#343) 
Co-authored-by: Alexander Soare <alexander.soare159@gmail.com>
 2024-07-26 12:47:03 +01:00
Alexander Soare f8a6574698
Add online training with TD-MPC as proof of concept (#338) 2024-07-25 11:16:38 +01:00
Alexander Soare abbb1d2367
Make sure policies don't mutate the batch (#323) 2024-07-22 20:38:33 +01:00
Simon Alibert 0b21210d72
Convert datasets to av1 encoding (#302) 2024-07-22 20:08:59 +02:00
Simon Alibert 461d5472d3
Fix visualize_image_transforms (#333) 2024-07-18 22:26:00 +02:00
Simon Alibert 8865e19c12
Fix datasets missing versions (#318) 2024-07-16 23:02:31 +02:00
Alexander Soare 5f5efe7cb9
Improve error message when attempting to overwrite a training output folder (#322) 2024-07-16 16:50:31 +01:00