896a11f60e
backup wip
2024-03-19 18:50:04 +00:00
ea17f4ce50
backup wip
2024-03-19 16:02:09 +00:00
88347965c2
revert dp changes, make act and tdmpc batch friendly
2024-03-18 19:18:21 +00:00
09ddd9bf92
Merge branch 'main' into user/alexander-soare/multistep_policy_and_serial_env
2024-03-18 18:27:50 +00:00
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
8e346b379d
switch between train and eval
2024-03-18 09:45:17 +00:00
bae7e7b41c
Merge remote-tracking branch 'origin/main' into user/alexander-soare/multistep_policy_and_serial_env
2024-03-15 14:06:53 +00:00
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
3124f71ebd
Merge remote-tracking branch 'origin/main' into user/alexander-soare/multistep_policy_and_serial_env
2024-03-15 14:04:23 +00:00
4ecfd17f9e
fix wandb artifact name and add disable option
2024-03-15 13:56:55 +00:00
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
b752833f3f
fix download
2024-03-15 13:19:18 +00:00
a45896dc8d
Merge remote-tracking branch 'origin/main' into user/alexander-soare/multistep_policy_and_serial_env
2024-03-15 13:05:35 +00:00
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
5805a7ffb1
small fix in type + comments
2024-03-15 12:44:52 +00:00
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
b10c9507d4
Small fix
2024-03-15 00:36:55 +00:00
a311d38796
Add aloha + improve readme
2024-03-15 00:30:11 +00:00
19730b3412
Add pusht on hf dataset (WIP)
2024-03-14 16:59:37 +00:00
a222c88c99
Merge branch 'user/alexander-soare/train_pusht' into user/alexander-soare/multistep_policy_and_serial_env
2024-03-14 16:06:21 +00:00
736bc969ca
Merge branch 'main' into user/alexander-soare/train_pusht
2024-03-14 16:05:11 +00:00
4822d63dbe
Merge branch 'main' into user/alexander-soare/multistep_policy_and_serial_env
2024-03-14 16:04:40 +00:00
ba91976944
wip: still needs batch logic for act and tdmp
2024-03-14 15:24:10 +00:00
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
8e856f1bf7
Update readme
2024-03-14 15:24:38 +01:00
8c2b47752a
Remove cuda env copy
2024-03-14 13:55:35 +01:00
f515cb6efd
Add dm-control
2024-03-14 13:42:03 +01:00
c3f8d14fd8
CI env fix
2024-03-14 13:29:27 +01:00
98484ac68e
ready for review
2024-03-12 21:59:01 +00:00
9512d1d2f3
Merge branch 'main' into user/alexander-soare/train_pusht
2024-03-12 19:41:27 +00:00
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
998dd2b874
info
2024-03-12 14:45:35 +00:00
7331df81d2
info
2024-03-12 14:44:23 +00:00
2c5d49cad5
Add sudo apt-get install -y libegl1-mesa-dev
2024-03-12 14:30:28 +00:00
5881eec376
Add DEVICE constant from LEROBOT_TESTS_DEVICE
2024-03-12 14:14:39 +00:00
29c73844b1
Update test.yml
2024-03-12 13:52:53 +00:00
f9258898ff
CI fix attempt
2024-03-12 12:24:10 +01:00
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
060bac7672
Merge pull request #23 from Cadene/kashif-patch-1
...
Update README.md
2024-03-12 10:11:42 +01:00
337208f28d
Update README.md
...
fix some readme typos
2024-03-12 10:05:05 +01:00
87fcc536f9
wip - still need to verify full training run
2024-03-11 18:45:21 +00:00
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
4449c06823
Add comment
...
Co-authored-by: Remi <re.cadene@gmail.com>
2024-03-11 19:06:22 +01:00
304355c917
Merge remote-tracking branch 'origin/main' into train_pusht
2024-03-11 15:37:37 +00:00
2a01487494
early training loss as expected
2024-03-11 13:34:04 +00:00
a94800fc8a
Merge pull request #20 from Cadene/user/aliberts/2024_03_11_license
...
Add license
2024-03-11 14:32:05 +01:00
a207b416b7
Typo fix
2024-03-11 14:20:05 +01:00
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
6d6c84b4a3
Remove entity from config
...
Co-authored-by: Remi <re.cadene@gmail.com>
2024-03-11 14:14:17 +01:00
772a826bf2
Add FOWM & ALOHA copyright notices
2024-03-11 13:54:20 +01:00
Alexander Soare
Remi
Simon Alibert
Simon Alibert
Kashif Rasul