This is a repository containing the code for the paper BAKU: An Efficient Transformer for Multi-Task Policy Learning.
In order to install the required dependencies, please follow the instructions provide here.
We have added the instructions for running BAKU on the LIBERO benchmark here. For access to the datasets for Meta-World, DMControl, and the real world xArm Kitchen, please send an email to the [email protected].
If you find this work useful, please cite the paper using the following bibtex:
@article{haldar2024baku,
title={BAKU: An Efficient Transformer for Multi-Task Policy Learning},
author={Haldar, Siddhant and Peng, Zhuoran and Pinto, Lerrel},
journal={arXiv preprint arXiv:2406.07539},
year={2024}
}