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

526 Commits

Author SHA1 Message Date
Simon Alibert e2618bd674 Add back full testing 2024-05-03 10:04:49 +02:00
Simon Alibert 6f11b0afaf Cleanup 2024-05-03 10:04:49 +02:00
Simon Alibert 1acfd61b88 Dial back to atol=1e-7 2024-05-03 10:04:49 +02:00
Simon Alibert 97ded04b07 Try atol=1e-8 2024-05-03 10:04:49 +02:00
Simon Alibert 82b5ffb250 Try rtol=0.1 2024-05-03 10:04:49 +02:00
Simon Alibert 90cb75acfd atol=1e-7 2024-05-03 10:04:49 +02:00
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
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