Director-pytorch

PyTorch implementation of the Director agent, based on the paper Deep Hierarchical Planning from Pixels.

Based on the DreamerV2 PyTorch implementation by RajGhugare19.

Some example results can be seen at W&B here, e.g. Sokoban.

Usage

Install Poetry:

$ pip install poetry

Install dependencies:

$ poetry install

Run an experiment:

$ python pomdb.py --config-name breakout