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

529 Commits

Author SHA1 Message Date
Cadene 764a7ad2c3 try 2024-05-06 00:49:33 +00:00
Cadene cf4b4c5a18 fix 2024-05-06 00:44:20 +00:00
Cadene 5a867e34af fix 2024-05-06 00:34:08 +00:00
Cadene bf790bb153 Improve examples, Add save_model by default, Other minor 2024-05-06 00:19:45 +00:00
Cadene b7114fad7e Improve examples, Add save_model by default, Other minor 2024-05-05 23:58:29 +00:00
Cadene b15ec1ff6d Improve examples, Add save_model by default, Other minor 2024-05-05 23:40:28 +00:00
Cadene 57065871a4 Merge remote-tracking branch 'origin/main' into user/aliberts/2024_05_04_release_cleanup 2024-05-05 20:11:34 +00:00
Cadene b11abaddf7 Improve 2024-05-05 18:06:02 +00:00
Adil Zouitine b0b97cb8ae Fix video position 2024-05-05 19:40:41 +02:00
Adil Zouitine 5d51a732f4 Move rerun video 2024-05-05 19:39:49 +02:00
Adil Zouitine 9a9245d093 Add rerun video 2024-05-05 19:33:41 +02:00
Remi 1dc0da58f3
Update README.md 2024-05-05 19:21:06 +02:00
Remi 6eaffbef1d
Add 25 real world datasets (static + mobile aloha) (#133) 2024-05-05 19:13:11 +02:00
Cadene 6829a1fee4 Add media (forgot) 2024-05-05 17:12:41 +00:00
Cadene 0caa33309a image_keys -> camera_keys, add visualization video and wandb image to README 2024-05-05 17:01:50 +00:00
Adil Zouitine 13bd0a6184 Update dataset upload instructions and add support for new raw formats 2024-05-05 17:39:09 +02:00
Simon Alibert 0d31872550 Format 2024-05-05 15:09:08 +02:00
Simon Alibert 41acc73b4a Add todo for train 2024-05-05 15:04:23 +02:00
Simon Alibert 17778fa3cb Merge remote-tracking branch 'origin/main' into user/aliberts/2024_05_04_release_cleanup 2024-05-05 14:47:49 +02:00
Simon Alibert dc56e9b930 Move diffusers 'get_scheduler' import 2024-05-05 14:47:14 +02:00
Simon Alibert 53d8f6b785 Cleanup eval example 2024-05-05 14:46:29 +02:00
Alexander Soare 6e6feae09e
Remove additional frame from eval render (#137) 2024-05-05 13:49:53 +02:00
Simon Alibert f4dce55b46 Merge remote-tracking branch 'origin/main' into user/aliberts/2024_05_04_release_cleanup 2024-05-05 13:10:48 +02:00
Alexander Soare f6fe162432
Rename Diffusion Policy model repository (#136) 2024-05-05 12:06:58 +01:00
Simon Alibert adf884eb78 Apply suggestions from code review 
Co-authored-by: Alexander Soare <alexander.soare159@gmail.com>
 2024-05-05 12:43:58 +02:00
Simon Alibert 58755561bd Update walkthrough items 2024-05-05 12:43:58 +02:00
Simon Alibert 7664ad8259 Fix pretrained_policy_path 2024-05-05 12:43:58 +02:00
Simon Alibert f23f5f977f Update readme for visualization & examples 2024-05-05 12:43:58 +02:00
Simon Alibert 0981350449 Enhance example 1 2024-05-05 12:43:58 +02:00
Simon Alibert 292da1f4fe Enhance PR template 2024-05-05 12:43:57 +02:00
Kashif Rasul 649087a274 Update README.md 
Co-authored-by: Simon Alibert <75076266+aliberts@users.noreply.github.com>
 2024-05-05 12:43:57 +02:00
Kashif Rasul b88a9a9e6e fixed typos and Hugging Face in Readmes 2024-05-05 12:43:57 +02:00
Alexander Soare f3bba0270d
Remove EMA model from Diffusion Policy (#134) 2024-05-05 11:26:12 +01:00
Alexander Soare d747195c57
Fix loops in eval.py (#128) 2024-05-04 15:21:10 +01:00
Simon Alibert c77633c38c
Add regression tests (#119) 
- Add `tests/scripts/save_policy_to_safetensor.py` to generate test artifacts
- Add `test_backward_compatibility to test generated outputs from the policies against artifacts
 2024-05-04 16:20:30 +02:00
Remi 19812ca470
Add dataset visualization with rerun.io (#131) 
Co-authored-by: Simon Alibert <75076266+aliberts@users.noreply.github.com>
 2024-05-04 16:07:14 +02:00
Simon Alibert c015252e20
Remove batch length restrictions in select_action (#123) 2024-05-04 15:33:55 +02:00
Alexander Soare bccee745c3
Refactor eval.py (#127) 2024-05-03 17:33:16 +01:00
Simon Alibert b7b69fcc3d
Publish environments (#120) 2024-05-03 16:25:45 +02:00
Remi b2cda12f87
Add video decoding to LeRobotDataset (#92) 2024-05-03 00:50:19 +02:00
Alexander Soare c1668924ab
Fix missing `policy.to(device)` in policy factory (#126) 2024-05-01 17:26:58 +01:00
Alexander Soare d1855a202a
Refactor TD-MPC (#103) 
Co-authored-by: Cadene <re.cadene@gmail.com>
Co-authored-by: Simon Alibert <75076266+aliberts@users.noreply.github.com>
 2024-05-01 16:40:04 +01:00
Alexander Soare a4891095e4
Use PytorchModelHubMixin to save models as safetensors (#125) 
Co-authored-by: Remi <re.cadene@gmail.com>
 2024-05-01 16:17:18 +01:00
Alexander Soare 01d5490d44
Follow `transformers` single file naming conventions (#124) 2024-05-01 13:09:42 +01:00
Alexander Soare 986583dc5c
Nest ACT model in ACT Policy (#122) 2024-04-30 17:17:40 +01:00
Alexander Soare 9d60dce6f3
Tidy up yaml configs (#121) 2024-04-30 16:08:59 +01:00
Remi e4e739f4f8
Refactor push_dataset_to_hub (#118) 2024-04-30 14:25:41 +02:00
Simon Alibert 2765877f28
Speed up CI, add more checks (#115) 
- Split pytest and end-to-end tests into separate jobs
- Add poetry check to ensure pyproject.toml and poetry.lock are in sync
- Add ruff format --diff to ensure style formatting is applied (fails if ruff would reformat anything)
 2024-04-29 23:05:55 +02:00
Adil Zouitine 1ec5f77f7c
Remove mask check key in XarmProcessor (#117) 2024-04-29 22:00:37 +02:00
Quentin Gallouédec 508bd92d03
Remove `update` method from the policy (#99) 
Co-authored-by: Alexander Soare <alexander.soare159@gmail.com>
 2024-04-29 12:27:58 +02:00