### Steps to train Rainbow (Reinforcement Learning)

The folder contains code to train an RL agent to play Space Invaders (an Atari game). We make use 
of Deep Q-Networks for Reinforcement Learning

### Dependencies
```adabelief-pytorch==0.0.5```\
Create a conda environment from `environment.yml` and use it.

### Command to run 

>sh run.sh

The above command trains the model using AdaBelief as the optimizer, to make use of 
Adam comment Line 47 and uncomment Line 48.

### Results
We provide results for Adam in ```results/adam_run``` folder and AdaBelief in 
```results/adabelief_run``` folder
