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

135 Commits

Author SHA1 Message Date
Simon Alibert 590b0eb48f Add require_x86_64_kernel 2024-05-03 10:04:49 +02:00
Simon Alibert 55ff23c252 rtol=0.5 2024-05-03 10:04:49 +02:00
Simon Alibert 2d11199320 rtol=1, atol=1e-6 for all asserts 2024-05-03 10:04:49 +02:00
Simon Alibert 8f54825b1f rtol=1, atol=1e-6 2024-05-03 10:04:49 +02:00
Simon Alibert 8e842ac00c rtol=1e-4, atol=1e-7 2024-05-03 10:04:49 +02:00
Simon Alibert e75448c8c4 Generate artifacts from x86 cpu 2024-05-03 10:04:49 +02:00
Simon Alibert 2ca594e801 atol = 1e-3 2024-05-03 10:04:49 +02:00
Simon Alibert ff2cc0d7b1 Increase tolerance further again 2024-05-03 10:04:49 +02:00
Simon Alibert 53ae422eb8 Increase tolerance further 2024-05-03 10:04:49 +02:00
Simon Alibert c3e57c2ae7 Increase tolerance 2024-05-03 10:04:49 +02:00
Simon Alibert a66031f1f7 Remove default params torch.isclose 2024-05-03 10:04:49 +02:00
Simon Alibert 14648f81d0 Update cpu-generated artifact 2024-05-03 10:04:49 +02:00
Simon Alibert 77ee567851 Add require_cpu 2024-05-03 10:04:49 +02:00
Simon Alibert 62c1b9149a Update artifacts 2024-05-03 10:04:49 +02:00
Simon Alibert 344c1653f2 Fix device 2024-05-03 10:04:49 +02:00
Simon Alibert 28d4122929 Create artifacts with DATA_DIR=tests/data 2024-05-03 10:04:49 +02:00
Simon Alibert 326933f208 Remove require_cuda 2024-05-03 10:04:49 +02:00
Simon Alibert b0db762236 Update test artifacts 2024-05-03 10:04:49 +02:00
Simon Alibert 5877eb38ab Enable backward compatibility testing on tdmpc 2024-05-03 10:04:49 +02:00
Simon Alibert 3a918b980f WIP 2024-05-03 10:04:48 +02:00
Simon Alibert 276d210380 Add require_cuda 2024-05-03 10:04:48 +02:00
Simon Alibert c1ecfdc663 Move to test_policies 2024-05-03 10:04:48 +02:00
Simon Alibert 30b32814a9 Add artifacts 2024-05-03 10:04:48 +02:00
Simon Alibert 7002813bbb Cleanup, add output_dict 2024-05-03 10:04:48 +02:00
Simon Alibert 7ee03afabb Test backward_compatibility 2024-05-03 10:04:48 +02:00
Simon Alibert 2726d1a6c9 Add get_policy_stats 2024-05-03 10:04:48 +02:00
Simon Alibert 6e15e9664b WIP Add save_policy_to_safetensors 2024-05-03 10:04:48 +02:00
Simon Alibert d84e37803f WIP add test_backward_compatibility 2024-05-03 10:04:48 +02:00
Simon Alibert 14f80f294b WIP Add test_backward 2024-05-03 10:04:48 +02:00
Remi b2cda12f87
Add video decoding to LeRobotDataset (#92) 2024-05-03 00:50:19 +02: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 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
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
Alexander Soare ccffa9e406
Fix policy defaults (#113) 2024-04-29 08:26:59 +01:00
Adil Zouitine 55dc9f7f51
Refactor the download and publication of the datasets and convert it into CLI script (#95) 
Co-authored-by: Remi <re.cadene@gmail.com>
 2024-04-29 00:08:17 +02:00
Adil Zouitine 81e490d46f
Add UMI-gripper dataset (#83) 
Co-authored-by: Remi <re.cadene@gmail.com>
 2024-04-28 18:41:07 +02:00
Simon Alibert bf2eebb090
Update readme & remove example 1 (#108) 
Co-authored-by: Remi <re.cadene@gmail.com>
- Update instructions for installing the library
- Remove deprecated example 1 (as we are now only using `LeRobotDataset` since #91)
 2024-04-27 09:48:02 +02:00
Simon Alibert fdf6a0c4e3
More CI cleanup, add style workflow (#107) 
- Changes on the `test.yml` workflow:
  - Using poetry instead of pip. Contrary to what I wrote in #75, it is possible to use poetry (and have the benefits of shorter install times) without the need for having two separate versions of `pyproject.toml` and `poetry.lock`.
  - Reduce the trigger scope to only run when files in these directories are modified:
    - `lerobot/`
    - `tests/`
    - `examples/`
    - `.github/`
- Add `style.yml` workflow for doing a `ruff check` pass on the code
- More cleanup (removed deprecated workflow)
 2024-04-27 09:37:56 +02:00
Simon Alibert b980c5dd9e
CI nightlies cpu/gpu & cleanup (#75) 2024-04-25 14:58:39 +02:00
Remi 659c69a1c0
Refactor datasets into LeRobotDataset (#91) 
Co-authored-by: Alexander Soare <alexander.soare159@gmail.com>
 2024-04-25 12:23:12 +02:00
Remi e760e4cd63
Move normalization to policy for act and diffusion (#90) 
Co-authored-by: Alexander Soare <alexander.soare159@gmail.com>
 2024-04-25 11:47:38 +02:00
Remi 1030ea0070
Loads episode_data_index and stats during dataset __init__ (#85) 
Co-authored-by: Simon Alibert <75076266+aliberts@users.noreply.github.com>
Co-authored-by: Alexander Soare <alexander.soare159@gmail.com>
 2024-04-23 14:13:25 +02:00
Simon Alibert 7ad1909641
Tests cleaning & simplification (#81) 2024-04-18 14:47:42 +02:00
Remi 0928afd37d
Improve dataset examples (#82) 
Co-authored-by: Alexander Soare <alexander.soare159@gmail.com>
 2024-04-18 11:43:16 +02:00
Alexander Soare bff4b673c9 Merge remote-tracking branch 'upstream/main' into unify_policy_api 2024-04-17 08:08:57 +01:00
Cadene 91badebdfc fix tests 2024-04-16 17:29:31 +00:00
Cadene 36d9e885ef Address comments 2024-04-16 17:20:54 +00:00