The multiworld environments are taken from https://github.com/vitchyr/multiworld/tree/leap
To train RIS on the U-shaped ant maze environment, please run:
python train_ant.py --env_name AntU
Use this table to run RIS on other ant maze navigation tasks:
| Environment | --env_name |
|---|---|
| U-shaped ant maze (default) | AntU |
| S-shaped ant maze | AntFb |
|
|
AntMaze |
|
|
AntFg |
To train RIS on the vision-based robotic manipulation environment, please run:
python train_sawyer.py