AUTHOR: Christopher Stanton
DATE: 23 May 2018

The 'baselines' folder contains all code for the version of OpenAI Baselines
used for implementing DeepCS.
To use DeepCS in an experiment, install this module via setup.py (enclosed).

The 'gym' folder instead contains one file (atari_env.py) that was modified
to implement DeepCS.
Copy this file into an existing OpenAI Gym installation.

You can train a new model via A2C or ACKTR using baselines_train.sh.
To evaluate a trained model (e.g. to generate a video), use baselines_test.sh.
A sample run file (mrCS.sh) is provided to help get you started.
Program parameters are described in baselines/baselines/a2c/run_atari.sh.

