Skip to content

Latest commit

 

History

History
29 lines (19 loc) · 733 Bytes

README.md

File metadata and controls

29 lines (19 loc) · 733 Bytes

Rainbow DQN

Implementation of Rainbow-DQN agent for state-of-the-art discrete action design space exploration.

Installation

Execute:

conda env create --file environment.yml
conda activate rainbow

Optional:

Upgrade PyTorch:

conda install pytorch torchvision torchaudio pytorch-cuda=12.1 -c pytorch -c nvidia

NOTES

Tested on gym-super-mario-bros environment.

TODO

  • Complete comments and docstrings
  • Add flexibility in environment for either 1D or 2D observations
  • Enrich this README (main page) with a description of the project, results, lab page, etc.