Requirements

See requirements.txt.

Instructions to run

For original antmaze-v2: (replacing the env name with any of the six antmaze-v2 envs)

cd ./cgo
python main.py --env-name "antmaze-umaze-diverse-v2"

For Four Rooms: (replacing the env name with any of the six antmaze-v2 envs)

cd ./cgo
python main_four_rooms.py --env-name "antmaze-medium-play-v2"

For Random Cells: (replacing the env name with any of the six antmaze-v2 envs)

cd ./cgo
python main_cell.py --env-name "antmaze-medium-play-v2"