/babyai_sr

A package that uses BabyAI in a sender/receiver setup.

Primary LanguagePythonBSD 3-Clause "New" or "Revised" LicenseBSD-3-Clause

BabyAI S/R

This is a package that uses BabyAI in a sender/receiver setup.

Installation

First, install this BabyAI fork and the corresponding MiniGrid fork, which ease compatibility. (If you’re feeling lucky, you can instead try installing the original BabyAI repo (and the corresponding original MiniGrid repo).)

Then, clone this repository and install it with pip3:

git clone https://github.com/thomasaunger/babyai_sr.git
cd babyai_sr
pip3 install --editable .

In order to use the plotting and visualization scripts as is, you’ll also need to install LaTeX.

Usage

This package is organized similarly to the original BabyAI repo. Models can be trained using scripts/train_rl.py.

The resulting models can be tested using scripts/test_rl.py, which also saves the episode data to an NPY file, which in turn can be visualized using the vis scripts.

The learning curves as described by the log files can be plotted using scripts/plot_curves.py.