Commit Graph

7 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 Cadene
570f8d01df Revert "WIP"
This reverts commit 20c08bb740.
2024-03-04 22:41:47 +00:00
Remi Cadene
20c08bb740 WIP 2024-03-04 22:37:08 +00:00
Simon Alibert
7e024fdce6 Ran pre-commit run --all-files 2024-02-29 13:37:48 +01:00
Cadene
ac90b9c3ee Fix diffusion (rm transpose), Add prefetch 2024-02-28 17:45:01 +00:00
Cadene
591985c67d Fix done in pusht, Fix --time in sbatch 2024-02-22 17:58:26 +00:00
Cadene
664cfb2023 Add sbatch.sh 2024-02-22 13:04:32 +00:00