Restraining-Bolts-for-Reinforcement-Learning

Installation

1. Clone this repository
2. Install dependencies

   pip3 install -r requirements.txt

3. Run setup from the repository root directory

   python3 setup.py install

Training

$ python3 train.py
    --episodes 10000
    --env
    --algo dqn
    --gui False
    --model_name
    --rand_seed 42

Problems and solutions of training Deep Neural Network for RB:deep_learning.md.

Overleaf source code for the presentation: https://www.overleaf.com/read/prtnncqwtszz

Testing

$ python3 test.py
    --model_name
    --record True
    --gui True

If you want to record, please remember to install ffmpeg

On linux:

sudo apt-get install ffmpeg

On ubuntu:

brew install ffmpeg

On Windows: good luck