Healthcare-Robotics / assistive-gym

Assistive Gym, a physics-based simulation framework for physical human-robot interaction and robotic assistance.
MIT License
301 stars 73 forks source link

Google Colab #7

Closed kavikode closed 3 years ago

kavikode commented 3 years ago

Is it possible to run it on the Google Colab because Python Bullet does not seem to work there since it cannot connect to the X-Server?

Zackory commented 3 years ago

Hi kavivode,

It is likely that you can train new assistive policies on Google Colab (see the Wiki: Training New Policies). Training a new policy does not require visualization or X-Server.

However, you will likely need Assistive Gym installed locally on a machine in order to visualize environments or trained policies.

For example, the command below won't work on Google Colab, since it requires brining up an interactive window through X-Server. python3 env_viewer.py --env "BedBathingSawyer-v0"

Zackory commented 3 years ago

However, it is possible to render a non-interactive simulation in Google Colab using an animated png. Here is an example, which generates an animated png from a pybullet environment.

https://colab.research.google.com/drive/1u6j7JOqM05vUUjpVp5VNk0pd8q-vqGlx?authuser=1#scrollTo=7tbOVtFp1_5K

Zackory commented 3 years ago

Assistive Gym now support Google Colab! Check out this Wiki Page for examples.