Skip to content

The reproduction code of the imitation learning algorithm Action Chunking with Transformer (ACT)

License

Notifications You must be signed in to change notification settings

HenryWJL/act_pytorch

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

82 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

This is the PyTorch implementation of Learning Fine-Grained Bimanual Manipulation with Low-Cost Hardware. Code in this repository is copied from the official codebase and simplified such that minimal requirements are demanded. Also, sufficient annotations are added to help users better understand the algorithm.

Installation

Create a conda environment:

conda env create -f conda_env.yaml
conda activate act

Usage

1. Collect Human Demonstrations.

python utils/collect_data.py

2. Train ACT Policy.

python train.py --dataset_dir <YOUR_DATASET_ROOT>

3. Evaluate ACT policy.

python eval.py --checkpoint <YOUR_CHECKPOINT_PATH>

About

The reproduction code of the imitation learning algorithm Action Chunking with Transformer (ACT)

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages