Commit Graph

170 Commits

Author SHA1 Message Date
Cadene 8bed0fc465 WIP 2024-03-20 10:10:00 +00:00
Remi 2d5abbbd6f
Merge pull request #36 from Cadene/user/rcadene/2024_03_19_replay_buffer_folder
Add replay_buffer directory and dataset versioning
2024-03-19 18:04:29 +01:00
Cadene 7d5d99e036 Address more comments 2024-03-19 16:53:07 +00:00
Cadene b420ab88f4 version naming conventions 2024-03-19 16:44:19 +00:00
Cadene e799dc5e3f Improve mock_dataset 2024-03-19 16:38:07 +00:00
Cadene 10034e85c4 Aloha done 2024-03-19 16:03:42 +00:00
Cadene 6a1a29386a Add replay_buffer directory in pusht datasets + aloha (WIP) 2024-03-19 15:49:45 +00:00
Remi 099a465367
Merge pull request #35 from Cadene/switch_between_train_and_eval
Switch between train and eval modes
2024-03-18 13:21:16 +01:00
Alexander Soare 8e346b379d switch between train and eval 2024-03-18 09:45:17 +00:00
Remi 75cc10198f
Merge pull request #31 from Cadene/disable_wandb_artifact
Fix wandb artifact name and add disable option
2024-03-15 15:06:16 +01:00
Alexander Soare 4ecfd17f9e fix wandb artifact name and add disable option 2024-03-15 13:56:55 +00:00
Remi 58d1787ee3
Merge pull request #30 from Cadene/user/rcadene/2024_03_15_fix_path
Use Path type instead of str for data_dir
2024-03-15 14:41:33 +01:00
Cadene b752833f3f fix download 2024-03-15 13:19:18 +00:00
Remi 9c88071bc7
Merge pull request #28 from Cadene/user/rcadene/2024_03_14_hf_dataset
Download datasets from hugging face
2024-03-15 13:52:13 +01:00
Cadene 5805a7ffb1 small fix in type + comments 2024-03-15 12:44:52 +00:00
Cadene 41521f7e96 self.root is Path or None + The following packages are already present in the pyproject.toml and will be skipped:
- huggingface-hub

If you want to update it to the latest compatible version, you can use `poetry update package`.
If you prefer to upgrade it to the latest available version, you can use `poetry add package@latest`.

Nothing to add.
2024-03-15 10:56:46 +00:00
Cadene b10c9507d4 Small fix 2024-03-15 00:36:55 +00:00
Cadene a311d38796 Add aloha + improve readme 2024-03-15 00:30:11 +00:00
Cadene 19730b3412 Add pusht on hf dataset (WIP) 2024-03-14 16:59:37 +00:00
Simon Alibert 95e84079ef
Merge pull request #25 from Cadene/user/aliberts/2024_03_13_ci_fix
CI env fix
2024-03-14 15:24:56 +01:00
Simon Alibert 8e856f1bf7 Update readme 2024-03-14 15:24:38 +01:00
Simon Alibert 8c2b47752a Remove cuda env copy 2024-03-14 13:55:35 +01:00
Simon Alibert f515cb6efd Add dm-control 2024-03-14 13:42:03 +01:00
Simon Alibert c3f8d14fd8 CI env fix 2024-03-14 13:29:27 +01:00
Remi 8c56770318
Merge pull request #11 from Cadene/user/rcadene/2024_03_06_aloha_env
[WIP] Add Aloha env and ACT policy
2024-03-12 16:02:48 +01:00
Cadene 998dd2b874 info 2024-03-12 14:45:35 +00:00
Cadene 7331df81d2 info 2024-03-12 14:44:23 +00:00
Cadene 2c5d49cad5 Add sudo apt-get install -y libegl1-mesa-dev 2024-03-12 14:30:28 +00:00
Cadene 5881eec376 Add DEVICE constant from LEROBOT_TESTS_DEVICE 2024-03-12 14:14:39 +00:00
Cadene 29c73844b1 Update test.yml 2024-03-12 13:52:53 +00:00
Simon Alibert f9258898ff CI fix attempt 2024-03-12 12:24:10 +01:00
Remi Cadene 9d002032d1 Add Aloha env and ACT policy
WIP Aloha env tests pass

Rendering works (fps look fast tho? TODO action bounding is too wide [-1,1])

Update README

Copy past from act repo

Remove download.py add a WIP for Simxarm

Remove download.py add a WIP for Simxarm

Add act yaml (TODO: try train.py)

Training can runs (TODO: eval)

Add tasks without end_effector that are compatible with dataset, Eval can run (TODO: training and pretrained model)

Add AbstractEnv, Refactor AlohaEnv, Add rendering_hook in env, Minor modifications, (TODO: Refactor Pusht and Simxarm)

poetry lock

fix bug in compute_stats for action normalization

fix more bugs in normalization

fix training

fix import

PushtEnv inheriates AbstractEnv, Improve factory Normalization

Add _make_env to EnvAbstract

Add call_rendering_hooks to pusht env

SimxarmEnv inherites from AbstractEnv (NOT TESTED)

Add aloha tests artifacts + update pusht stats

fix image normalization: before env was in [0,1] but dataset in [0,255], and now both in [0,255]

Small fix on simxarm

Add next to obs

Add top camera to Aloha env (TODO: make it compatible with set of cameras)

Add top camera to Aloha env (TODO: make it compatible with set of cameras)
2024-03-12 10:27:48 +00:00
Remi 060bac7672
Merge pull request #23 from Cadene/kashif-patch-1
Update README.md
2024-03-12 10:11:42 +01:00
Kashif Rasul 337208f28d
Update README.md
fix some readme typos
2024-03-12 10:05:05 +01:00
Simon Alibert 48e70e044e
Merge pull request #18 from Cadene/user/aliberts/2024_03_11_dynamic_build_versioning
Add dynamic build versioning
2024-03-11 19:06:51 +01:00
Simon Alibert 4449c06823
Add comment
Co-authored-by: Remi <re.cadene@gmail.com>
2024-03-11 19:06:22 +01:00
Simon Alibert a94800fc8a
Merge pull request #20 from Cadene/user/aliberts/2024_03_11_license
Add license
2024-03-11 14:32:05 +01:00
Simon Alibert a207b416b7 Typo fix 2024-03-11 14:20:05 +01:00
Simon Alibert 78690d197f
Merge pull request #19 from Cadene/user/aliberts/2024_03_11_wandb_config
Configure wandb entity outside config
2024-03-11 14:17:44 +01:00
Simon Alibert 6d6c84b4a3
Remove entity from config
Co-authored-by: Remi <re.cadene@gmail.com>
2024-03-11 14:14:17 +01:00
Simon Alibert 772a826bf2 Add FOWM & ALOHA copyright notices 2024-03-11 13:54:20 +01:00
Simon Alibert 2cb8ae5037 Add acknowledgements 2024-03-11 13:49:08 +01:00
Remi fab2b3240b
Merge pull request #17 from Cadene/user/rcadene/2024_03_11_bugfix_compute_stats
Fix bugs with normalization
2024-03-11 13:44:07 +01:00
Cadene 84a1647c01 fix import 2024-03-11 12:41:14 +00:00
Cadene ccd5dc5a42 fix training 2024-03-11 12:33:15 +00:00
Simon Alibert c1e9c13ade Add license 2024-03-11 13:31:52 +01:00
Simon Alibert 00fe4f4f18 Configure wandb entity outside config 2024-03-11 13:09:46 +01:00
Simon Alibert 225eebde40 Add dynamic build versioning 2024-03-11 12:10:46 +01:00
Cadene 816b2e9d63 fix more bugs in normalization 2024-03-11 11:03:51 +00:00
Cadene a7ef4a6a33 fix bug in compute_stats for action normalization 2024-03-11 09:47:54 +00:00