act-plus-plus icon indicating copy to clipboard operation
act-plus-plus copied to clipboard

Advice for running on other robot ?

Open apirrone opened this issue 1 year ago • 2 comments

Hi @MarkFzp @tonyzhaozh,

I managed to run yout code and train a policy from sim_transfer_cube_scripted (It is not really working, but it just probably need some more training steps)

I would like to run mobile-aloha on a different mobile bimanual robot (Reachy https://www.pollen-robotics.com/)

Could you give me an overview of the required steps ?

Do I need to describe the new robot in the mujoco format ? I am not sure I understand properly what is the role of the simulated environment in your training pipeline. Reading the paper, I did not really understant how you would specify a task like "Wipe wine" in the simulated environment.

Anyways, I guess I need to record episodes of a teleoperated task, and format it like in https://github.com/MarkFzp/mobile-aloha/blob/main/aloha_scripts/record_episodes.py ?

Thank you very much for your help !

Antoine

apirrone avatar Jan 05 '24 10:01 apirrone

I was able to adapt the code to train and run with my robot :)

I can train policies from scratch based on recordings of teleoperation sessions. I haven't trained it for a long time yet but it seems promising !

Now I would like to perform "co-training" as described in the paper. Where can I download the dataset ? And how do I use it along with a different robot architecture ?

Thanks !

apirrone avatar Jan 05 '24 18:01 apirrone

I also briefly dig into the code, just to comment here if people would like to know the answer.

  • to run on another robot, user needs to create a custom robot env. Refer to here. In this case this script uses aloha_scripts from the mobile aloha repo.
  • I might be wrong, the sim env mainly works as a CI or demo example to run the pipeline without the real robot. It uses a scripted policy to generate a traj dataset, then it showcases a simple BC training.
  • Refered to the paper, cotraining trains along with the static aloha arms dataset. The dataset has already been provided, mentioned on the website.

youliangtan avatar Jan 09 '24 20:01 youliangtan

Thanks for the response @youliangtan , you are absolutely correct.

The sim envs are just legacy envs from the original ACT code (https://github.com/tonyzhaozh/act). They do not have the mobile component and we are not using them actively in the mobile-aloha project.

tonyzhaozh avatar Jan 11 '24 05:01 tonyzhaozh

@tonyzhaozh hello, I also want to know, if my robot has different architecture(such as arms), can I use the co-training with the static ALOHA data? Thanks

uuu686 avatar Jan 11 '24 08:01 uuu686

@apirrone hi bro, could you give some details to run the code on my own robot,thanks a lot.

zxfever avatar Apr 19 '24 07:04 zxfever