dqn-in-the-caffe icon indicating copy to clipboard operation
dqn-in-the-caffe copied to clipboard

An implementation of Deep Q-Network using Caffe

DQN-in-the-Caffe

DQN-in-the-Caffe is an implementation of Deep Q-Network using Caffe.

Requirements

  • Caffe (with two pull requests and minor fixes on them)
  • https://github.com/BVLC/caffe/pull/1228 for step execution of solving
  • https://github.com/BVLC/caffe/pull/1122 for AdaDelta solver
  • At the moment my forked repository's dqn branch https://github.com/muupan/caffe/tree/dqn can be used with DQN-in-the-Caffe
  • Arcade Learning Environment
  • etc.

Algorithm Details

See http://www.cs.toronto.edu/~vmnih/docs/dqn.pdf for the details of DQN.

Demo

https://www.youtube.com/watch?v=p88R2_3yWPA