Commit Graph

518 Commits

Author SHA1 Message Date
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
Simon Alibert a541a7f3cf Add make_optimizer 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 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
Adil Zouitine 5b4fd8891d
Update UmiProcessor default fps to 10 (#116) 2024-04-29 12:11:34 +02:00
Alexander Soare ccffa9e406
Fix policy defaults (#113) 2024-04-29 08:26:59 +01:00
Simon Alibert 791506dfb8
Remove warnings (#111)
- Replace `use_pretrained_backbone` with `pretrained_backbone_weights`
- Bump diffusers' minimum version `0.26.3` -> `0.27.2`
- Add ignore flags in CI's pytest
- Change Box observation spaces in simulation environments
- Set `version_base="1.2"` in Hydra initializations
- Bump einops' minimum version `0.7.0` -> `0.8.0`
2024-04-29 00:31:33 +02: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 a4b6c5e3b1
Add torchaudio (#110) 2024-04-27 14:00:38 +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 fe2b9af64f
Add test-docker-build workflow (#109)
Adds a `test-docker-build.yml` workflow to ensure that docker images can correctly build when their Dockerfile has been modified on PRs.
2024-04-27 09:40:21 +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
Alexander Soare 45f351c618
Make sure targets are normalized too (#106) 2024-04-26 11:18:39 +01:00
Simon Alibert b980c5dd9e
CI nightlies cpu/gpu & cleanup (#75) 2024-04-25 14:58:39 +02:00