Skip to content

Simple DQN implementation for Atari's discrete action space. I use Vectorized environments for batch processing and more efficient replay buffers. Algorithm from: https://www.nature.com/articles/nature14236.

Notifications You must be signed in to change notification settings

vishwassathish/dqn_vectorized_env

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

4 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Simple DQN implementation for Atari's discrete action space.
Algorithm from: https://www.nature.com/articles/nature14236.

Simply run python train.py.

  • The agent should learn to play Breakout in about 2 hrs.
  • Expert level performance takes a few more hours.
  • Also provided trained model checkpoints for breakout and pong after about a million gradient steps

Code by Vishwas Sathish
Code references:

About

Simple DQN implementation for Atari's discrete action space. I use Vectorized environments for batch processing and more efficient replay buffers. Algorithm from: https://www.nature.com/articles/nature14236.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published