MarkFzp / act-plus-plus

Imitation learning algorithms with Co-training for Mobile ALOHA: ACT, Diffusion Policy, VINN
https://mobile-aloha.github.io/
MIT License
2.86k stars 525 forks source link

Add Docker support #32

Open rubendsa opened 5 months ago

rubendsa commented 5 months ago

Adds adds a Dockerfile based on a Pytorch base image along with some simple documentation to help deploy simulated data and training in different host environments.

Tested remotely on an L4 GCP instance running Ubuntu 20.04 and on a 4090 box running Ubuntu 22.04.