Rainbow: Combining Improvements in Deep Reinforcement Learning
- Follow setup instructions at https://github.com/openai/gym to install openai gym and atari games.
pip install gym
pip install -e '.[atari]'
- install pytorch and torchvision:
conda install pytorch torchvision -c pytorch
- install tensorflow and tensorboardX for logging.
pip install tensorboard
pip install tensorboardX
- DQN tutorial
- Vanilla DQN agent class
To train an agent on breakout for 2000000 steps.
python train.py --env Breakout-v0 --num_steps 2000000 --agent_name breakoutdqn
To watch your agent play 20 episodes of breakout.
python play.py --env Breakout-v0 --num_episodes 20 --agent_name breakoutdqn