Skip to content

Commit

Permalink
README updates (#4)
Browse files Browse the repository at this point in the history
  • Loading branch information
lukeschmitt-tr authored May 23, 2024
1 parent 7edb35e commit 7e15382
Showing 1 changed file with 11 additions and 10 deletions.
21 changes: 11 additions & 10 deletions README.md
Original file line number Diff line number Diff line change
@@ -1,27 +1,28 @@
# ACT: Action Chunking with Transformers

Project Websites:
Original Project Websites:

* [ALOHA](https://tonyzhaozh.github.io/aloha/)
* [Mobile ALOHA](https://mobile-aloha.github.io/)

This repo contains the implementation of ACT, together with 2 simulated environments:

* Transfer Cube
* Bimanual Insertion.
* Bimanual Insertion

You can train and evaluate ACT in simulation or on real hardware.
For real hardware, you would also need to install [ALOHA](https://github.com/Interbotix/aloha).

# Repo Structure

* act
* act
* ``detr`` Model definitions of ACT, modified from [DETR](https://github.com/facebookresearch/detr)
* ``policy.py`` An adaptor for ACT policy
* ``sim_env.py`` Mujoco + DM_Control environments with joint space control
* ``ee_sim_env.py`` Mujoco + DM_Control environments with EE space control
* ``scripted_policy.py`` Scripted policies for sim environments
* ``constants.py`` Constants shared across files
* ``utils.py`` Utils such as data loading and helper functions
* ``detr`` Model definitions of ACT, modified from [DETR](https://github.com/facebookresearch/detr)
* ``policy.py`` An adaptor for ACT policy
* ``sim_env.py`` Mujoco + DM_Control environments with joint space control
* ``ee_sim_env.py`` Mujoco + DM_Control environments with EE space control
* ``scripted_policy.py`` Scripted policies for sim environments
* ``constants.py`` Constants shared across files
* ``utils.py`` Utils such as data loading and helper functions
* scripts
* ``imitate_episodes.py`` Train and Evaluate ACT
* ``record_sim_episodes.py`` Record episodes using the simulator
Expand Down

0 comments on commit 7e15382

Please sign in to comment.