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

..

data/pusht

test ci

2024-03-09 15:41:23 +01:00

scripts

Add mock_dataset.py

2024-03-09 15:36:20 +01:00

__init__.py

Rename test -> tests

2024-02-25 10:51:07 +00:00

test_datasets.py

Add Aloha env and ACT policy

2024-03-10 22:06:12 +00:00

test_envs.py

Add Aloha env and ACT policy

2024-03-10 22:06:12 +00:00

test_policies.py

Refactor train, eval_policy, logger, Add diffusion.yaml (WIP)

2024-02-26 01:10:09 +00:00

utils.py

Refactor configs to have env in seperate yaml + Fix training

2024-02-25 17:42:47 +00:00