vwxyzjn / enn-zoo

Collection of entity-gym bindings for different reinforcement learning environments.

Entity Neural Network Zoo

The enn-zoo package collects entity-gym bindings for a number of reinforcement learning environments:

• Procgen
• Griddly
• MicroRTS
• ViZDoom
• CodeCraft

Setup

git clone https://github.com/entity-neural-network/enn-zoo.git
cd enn-zoo
poetry install
poetry run pip install setuptools==59.5.0
poetry run pip install torch==1.10.2+cu113 -f https://download.pytorch.org/whl/cu113/torch_stable.html
poetry run pip install torch-scatter -f https://data.pyg.org/whl/torch-1.10.0+cu113.html

Some of the environments have additional dependencies:

sudo apt install python3-dev make build-essential libssl-dev zlib1g-dev \
libbz2-dev libreadline-dev libsqlite3-dev wget curl llvm \
libncursesw5-dev xz-utils tk-dev libxml2-dev libxmlsec1-dev libffi-dev liblzma-dev

Usage

poetry run enn_zoo/train.py