8357dae26a
first draft
2024-05-08 12:42:40 +01:00
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
6e6feae09e
Remove additional frame from eval render ( #137 )
2024-05-05 13:49:53 +02:00
f6fe162432
Rename Diffusion Policy model repository ( #136 )
2024-05-05 12:06:58 +01:00
f3bba0270d
Remove EMA model from Diffusion Policy ( #134 )
2024-05-05 11:26:12 +01:00
d747195c57
Fix loops in eval.py ( #128 )
2024-05-04 15:21:10 +01:00
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
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
bccee745c3
Refactor eval.py ( #127 )
2024-05-03 17:33:16 +01:00
b2cda12f87
Add video decoding to LeRobotDataset ( #92 )
2024-05-03 00:50:19 +02:00
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
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
9d60dce6f3
Tidy up yaml configs ( #121 )
2024-04-30 16:08:59 +01:00
e4e739f4f8
Refactor push_dataset_to_hub ( #118 )
2024-04-30 14:25:41 +02:00
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
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
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
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
45f351c618
Make sure targets are normalized too ( #106 )
2024-04-26 11:18:39 +01:00
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
c1bcf857c5
Fix online training ( #94 )
2024-04-23 18:54:55 +02:00
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
e2168163cd
Quality of life patches for eval.py ( #86 )
2024-04-19 12:33:47 +01:00
7ad1909641
Tests cleaning & simplification ( #81 )
2024-04-18 14:47:42 +02:00
0928afd37d
Improve dataset examples ( #82 )
...
Co-authored-by: Alexander Soare <alexander.soare159@gmail.com>
2024-04-18 11:43:16 +02:00
bff4b673c9
Merge remote-tracking branch 'upstream/main' into unify_policy_api
2024-04-17 08:08:57 +01:00
91badebdfc
fix tests
2024-04-16 17:29:31 +00:00
36d9e885ef
Address comments
2024-04-16 17:20:54 +00:00
b241ea46dd
move download_and_upload_dataset.py to root_dir
2024-04-16 17:20:53 +00:00
e09d25267e
fix online training
2024-04-16 17:20:53 +00:00
4a3eac4743
fix unit tests, stats was missing, visualize_dataset was broken
2024-04-16 17:20:53 +00:00
0980fff6cc
HF datasets works
2024-04-16 17:19:40 +00:00
c7a8218620
typo
2024-04-16 17:19:39 +00:00
67d79732f9
Add download_and_upload_dataset.py in script, update all datasets, update online training
2024-04-16 17:19:39 +00:00
23be5e1e7b
backup wip
2024-04-16 16:31:44 +01:00
9241b5e830
pass step as kwarg
2024-04-15 09:52:54 +01:00
6d0a45a97d
ready for review
2024-04-12 11:36:52 +01:00
94cc22da9e
Merge remote-tracking branch 'upstream/main' into refactor_dp
2024-04-11 17:52:10 +01:00
976a197f98
backup wip
2024-04-11 17:51:35 +01:00
8e5b4365ac
fix
2024-04-11 13:57:22 +00:00
949f4d1a5b
remove comment
2024-04-10 17:21:36 +00:00
4c3d8b061e
Update lerobot/scripts/eval.py
...
Co-authored-by: Simon Alibert <75076266+aliberts@users.noreply.github.com>
2024-04-10 18:07:27 +02:00
8866b22db1
remove policy is None eval end-to-end tests
2024-04-10 15:09:04 +00:00
693f620df0
drop_last=False
2024-04-10 14:59:54 +00:00
48ec479660
fix end-to-end aloha
2024-04-10 14:26:30 +00:00
a18bcb39a7
cfg.env.fps
2024-04-10 14:02:11 +00:00
c08003278e
test_examples are passing
2024-04-10 13:45:45 +00:00
7c8eb7ff19
Merge remote-tracking branch 'origin/user/rcadene/2024_03_31_remove_torchrl' into user/rcadene/2024_03_31_remove_torchrl
2024-04-10 11:34:51 +00:00
06573d7f67
online training works (loss goes down), remove repeat_action, eval_policy outputs episodes data, eval_policy uses max_episodes_rendered
2024-04-10 11:34:01 +00:00
e6c6c2367f
Merge remote-tracking branch 'upstream/user/rcadene/2024_03_31_remove_torchrl' into refactor_act
2024-04-09 08:36:28 +01:00
Alexander Soare
Simon Alibert
Remi
Quentin Gallouédec
Adil Zouitine