Commit Graph

121 Commits

Author SHA1 Message Date
Remi Cadene
66323dcd0a 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)
2024-03-10 22:06:12 +00:00
Remi
d4ea4f0ad1 Merge pull request #16 from Cadene/user/aliberts/2024_03_09_integrate_diffusion_policy
Integrate diffusion policy
2024-03-10 17:02:16 +01:00
Simon Alibert
f54ee7cda0 Fix paths 2024-03-10 16:51:50 +01:00
Simon Alibert
134009f337 Remove init files 2024-03-10 16:38:49 +01:00
Simon Alibert
7982425670 Remove diffusion-policy dependency 2024-03-10 16:36:30 +01:00
Simon Alibert
6c867d78ef Integrate pusht env from diffusion 2024-03-10 16:33:03 +01:00
Simon Alibert
302b78962c Integrate diffusion policy 2024-03-10 15:31:17 +01:00
Simon Alibert
59397fb44a Move tdmpc files 2024-03-09 18:44:36 +01:00
Simon Alibert
1cc621ec36 Merge pull request #15 from Cadene/user/aliberts/2024_03_09_add_ci_eval_test
Add end-to-end pusht eval test
2024-03-09 16:54:50 +01:00
Simon Alibert
471ebfef62 Remove cleanup + reduce eval steps 2024-03-09 16:50:17 +01:00
Simon Alibert
30753d879c Fix 2024-03-09 16:41:31 +01:00
Simon Alibert
c6fb40fb29 Add end-to-end pusht eval test 2024-03-09 16:37:34 +01:00
Simon Alibert
fa7a947acc Merge pull request #12 from Cadene/user/aliberts/2024_03_08_test_data
Add pusht test artifact
2024-03-09 16:00:39 +01:00
Simon Alibert
450e32e4b5 update readme 2024-03-09 15:57:29 +01:00
Simon Alibert
0da85b2cef Removed debugging lines 2024-03-09 15:51:42 +01:00
Simon Alibert
f2c7ab5b3b test ci 2024-03-09 15:41:23 +01:00
Cadene
cde866dac0 reupload 2024-03-09 15:36:20 +01:00
Cadene
a54a0feb63 remove mock 2024-03-09 15:36:20 +01:00
Cadene
f440a681ad Add mock_dataset.py 2024-03-09 15:36:20 +01:00
Cadene
35bd577deb Add mock_dataset.py 2024-03-09 15:36:20 +01:00
Simon Alibert
327f60e4be Update readme 2024-03-09 15:36:20 +01:00
Simon Alibert
74ad9d5154 Change pusht artifacts path 2024-03-09 15:36:20 +01:00
Simon Alibert
89eaab140b Add pusht test artifact 2024-03-09 15:36:20 +01:00
Simon Alibert
7dbdbb051c Add git lfs in CI checkout 2024-03-09 15:35:20 +01:00
Simon Alibert
4cc7e1539e Merge pull request #10 from Cadene/user/rcadene/2024_03_06_fix_tests
Fix env tests
2024-03-08 12:30:08 +01:00
Simon Alibert
f1e2837d63 fix pusht data_dir path 2024-03-08 12:26:15 +01:00
Simon Alibert
54b05bfb77 Comment aloha, add test to CI 2024-03-07 15:57:38 +01:00
Remi Cadene
524d29aa80 fix tests 2024-03-07 13:23:22 +01:00
Simon Alibert
c2c0ef9927 Leaner caching strategy 2024-03-07 13:00:47 +01:00
Remi
66373e9b13 Merge pull request #9 from Cadene/user/rcadene/2024_03_05_aloha_dataset
Add Aloha replay buffer + 4 sim datasets
2024-03-06 14:51:55 +01:00
Simon Alibert
7d33b437fa Add h5py 2024-03-06 12:28:20 +01:00
Simon Alibert
b9dc3be463 Trigger CI on direct pushes 2024-03-06 12:13:23 +01:00
Simon Alibert
86ec62f98a Trigger CI on PR labeled CI 2024-03-06 12:02:01 +01:00
Simon Alibert
52bdfc659e Trigger CI on PR sync and reopen 2024-03-06 11:35:19 +01:00
Remi Cadene
d782b029e1 Add aloha dataset 2024-03-06 10:26:32 +00:00
Remi
49c0955f97 Merge pull request #7 from Cadene/user/rcadene/2024_03_05_abstract_replay_buffer
Add AbstractReplayBuffer
2024-03-06 11:25:24 +01:00
Remi Cadene
eed24b083a small fix 2024-03-06 10:21:22 +00:00
Remi Cadene
f95ecd66fc Improve visualize_dataset, Improve AbstractReplayBuffer, Small improvements 2024-03-06 10:15:57 +00:00
Simon Alibert
d34c0a3c49 Update CI caches 2024-03-05 17:19:39 +01:00
Simon Alibert
11a5a7ca45 Merge pull request #8 from Cadene/user/aliberts/2024_03_05_ci
Add CI, Remove old install files
2024-03-05 17:15:48 +01:00
Simon Alibert
a6d353c419 Fix 2024-03-05 17:00:17 +01:00
Simon Alibert
d6556e6519 Fix 2024-03-05 16:40:05 +01:00
Simon Alibert
12af67066d Fix 2024-03-05 16:26:31 +01:00
Simon Alibert
7a20ef65f6 Add CI, Remove old install files 2024-03-05 16:03:03 +01:00
Remi Cadene
2f80d71c3e Remove noqa-F821 2024-03-05 10:22:21 +00:00
Remi Cadene
d4e0849970 Refactor datasets with abstract class 2024-03-05 10:20:57 +00:00
Remi Cadene
e132a267aa offline_prioritized_sampler: true 2024-03-04 23:17:59 +00:00
Remi Cadene
a027f4edfb Add cfg.offline_prioritized_sampler 2024-03-04 23:08:52 +00:00
Remi Cadene
570f8d01df Revert "WIP"
This reverts commit 20c08bb740.
2024-03-04 22:41:47 +00:00
Remi Cadene
7938adcdfc Add sbatch_hopper.sh 2024-03-04 22:41:31 +00:00