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

545 Commits

Author SHA1 Message Date
Remi Cadene e1337aec71 pre-commit run --all-files 2024-05-19 19:56:06 +00:00
Remi Cadene a5098d7bf3 update aloha 2024-05-19 19:43:21 +00:00
Remi Cadene 155b3fdcf2 update aloha_act 2024-05-19 19:30:20 +00:00
Remi Cadene ed50e519bc require cpu for test_policies 2024-05-19 19:08:32 +00:00
Remi Cadene 780bf5d130 Add lfs: true for minimal test 2024-05-19 19:05:52 +00:00
Remi Cadene 923cd206fb add lfs: true 2024-05-19 18:49:14 +00:00
Remi Cadene 62ec7227e1 revert tests/data/save_policy_to_safetensors/ 2024-05-19 17:53:42 +00:00
Remi Cadene ab05c8b271 Merge remote-tracking branch 'origin/main' into user/rcadene/2024_05_11_fix_aloha_static_mobile_datasets 2024-05-19 17:46:37 +00:00
Remi Cadene ce499e506d tests/data/save_dataset_to_safetensors 2024-05-19 17:41:49 +00:00
Remi Cadene 28d72b9954 Update aloha_sim* 2024-05-19 17:27:05 +02:00
Remi Cadene db7d12bf64 Update aloha_sim_*_image 2024-05-19 16:51:50 +02:00
Remi Cadene 2d0012e663 Add tests aloha_mobile aloha_static 2024-05-19 15:28:28 +02:00
Remi Cadene 58f69d203b git lfs track *.json *.arrow *.mp4 2024-05-19 15:24:18 +02:00
Remi Cadene 52938d2b72 Fix episode_index in aloha datasets, Add velocity and effort to real aloha 2024-05-17 22:07:59 +00:00
Alexander Soare 096149b118
Disable wandb by default (#195) 2024-05-17 18:01:39 +01:00
Remi e0d851f7ce
Merge branch 'main' into user/rcadene/2024_05_11_fix_aloha_static_mobile_datasets 2024-05-17 17:10:50 +02:00
Remi Cadene 729e2e550b Minor improvements in push_dataset_to_hub 2024-05-17 14:50:55 +00:00
Remi Cadene c6fe57ede6 Add _image datasets in tests/data 2024-05-17 14:50:13 +00:00
Alexander Soare 5ec0af62c6
Explain why n_encoder_layers=1 (#193) 2024-05-17 15:05:40 +01:00
Alexander Soare 625f0557ef
Act temporal ensembling (#186) 2024-05-17 14:57:49 +01:00
Remi Cadene 7c5bd77d9c poetry lock --no-update 2024-05-17 02:13:03 +00:00
Remi Cadene 1d9433b411 WIP 
Add hf-transfer, num_workers 8

gc collect, rmtree save_to_disk False

Track .safetensors in git lfs

run save_dataset_to_safetensors on all datasets

Update datasets

Update download_raw
 2024-05-17 02:11:28 +00:00
Alexander Soare 4d7d41cdee
Fix act action queue (#185) 2024-05-16 15:43:25 +01:00
Akshay Kashyap c9069df9f1
Port SpatialSoftmax and remove Robomimic dependency (#182) 
Co-authored-by: Alexander Soare <alexander.soare159@gmail.com>
 2024-05-16 15:34:10 +01:00
Alexander Soare 68c1b13406
Make policies compatible with other/multiple image keys (#149) 2024-05-16 13:51:53 +01:00
Simon Alibert f52f4f2cd2
Add copyrights (#157) 2024-05-15 12:13:09 +02:00
Simon Alibert 89c6be84ca
Limit datasets major update (#176) 
Co-authored-by: Quentin Lhoest <42851186+lhoestq@users.noreply.github.com>
Co-authored-by: Quentin Lhoest <lhoest.q@gmail.com>
 2024-05-12 08:15:07 +02:00
AshisGhosh fc5cf3d84a
Fixes issue #152 - error with creating wandb artifact (#172) 
Co-authored-by: Ashis Ghosh <ahsisghosh@live.com>
Co-authored-by: Simon Alibert <75076266+aliberts@users.noreply.github.com>
 2024-05-12 08:13:12 +02:00
Simon Alibert 29a196c5dd
Fix #173 - Require gym-pusht to be installed for test_examples_3_and_2 (#174) 2024-05-12 08:08:59 +02:00
Remi ced3de4c94
Fix hanging in visualize_dataset.py when num_workers > 0 (#165) 2024-05-11 19:28:22 +03:00
Vincent Moens 7b47ab211b
Remove torchrl acknowledgement (#177) 2024-05-11 14:45:51 +03:00
Alexander Soare 1249aee3ac
Enable logging all the information returned by the `forward` methods of policies (#151) 2024-05-10 07:45:32 +01:00
Alexander Soare b187942db4
Add context manager for seeding (#164) 2024-05-09 17:58:39 +01:00
Alexander Soare 473345fdf6
Fix stats override in ACT config (#161) 2024-05-09 15:16:47 +01:00
Alexander Soare e89521dfa0
Enable tests for TD-MPC (#160) 2024-05-09 13:42:12 +01:00
Simon Alibert 7bb5b15f4c
Remove dependencies upper bounds constraints (#145) 2024-05-08 17:23:10 +00:00
Simon Alibert df914aa76c
Update dev docker build (#148) 2024-05-08 17:21:58 +00:00
Ikko Eltociear Ashimine 0ea7a8b2a3
refactor: update configuration_tdmpc.py (#153) 
Co-authored-by: Alexander Soare <alexander.soare159@gmail.com>
 2024-05-08 18:13:51 +01:00
Akshay Kashyap 460df2ccea
Support for DDIMScheduler in Diffusion Policy (#146) 2024-05-08 18:05:16 +01:00
Alexander Soare f5de57b385
Fix SpatialSoftmax input shape (#150) 2024-05-08 14:57:29 +01:00
Alexander Soare 47de07658c
Override pretrained model config (#147) 2024-05-08 12:56:21 +01:00
Alexander Soare 26d9a070d8
Add policy and dataset tags to WandB logging (#144) 2024-05-07 09:09:21 +01:00
Simon Alibert eb6bfe01b2
Relax version contraints & poetry lock (#139) 2024-05-06 17:38:29 +02:00
Alexander Soare a8e245fb31
Remove loss masking from diffusion policy (#135) 2024-05-06 07:27:01 +01:00
Simon Alibert f5e76393eb
Release cleanup (#132) 
Co-authored-by: Kashif Rasul <kashif.rasul@gmail.com>
Co-authored-by: Alexander Soare <alexander.soare159@gmail.com>
Co-authored-by: Adil Zouitine <adilzouitinegm@gmail.com>
Co-authored-by: Cadene <re.cadene@gmail.com>
 2024-05-06 03:03:14 +02:00
Remi 6eaffbef1d
Add 25 real world datasets (static + mobile aloha) (#133) 2024-05-05 19:13:11 +02:00
Alexander Soare 6e6feae09e
Remove additional frame from eval render (#137) 2024-05-05 13:49:53 +02:00
Alexander Soare f6fe162432
Rename Diffusion Policy model repository (#136) 2024-05-05 12:06:58 +01: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