Commit Graph

15 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
Simon Alibert
f54ee7cda0 Fix paths 2024-03-10 16:51:50 +01:00
Simon Alibert
6c867d78ef Integrate pusht env from diffusion 2024-03-10 16:33:03 +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
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
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
Cadene
21670dce90 Refactor train, eval_policy, logger, Add diffusion.yaml (WIP) 2024-02-26 01:10:09 +00:00
Cadene
b16c334825 Refactor configs to have env in seperate yaml + Fix training 2024-02-25 17:42:47 +00:00
Cadene
6f5c731936 Rename test -> tests 2024-02-25 10:51:07 +00:00