/DQN

A clean and minimal implementation of DQN (Deep Q Learning) algorithm in Pytorch.

Primary LanguagePython

Deep Q Learning (DQN)

This repository contains a clean and minimal implementation of Deep Q Learning (DQN) algorithm in Pytorch.

Tested on

  • Cart Pole (OpenAI Gym) - Move back and forth to balance a pole on a cart.

  • Mountain Car (OpenAI Gym) - Move back and forth to build up enough momentum to drive up a big hill.

References

  • Mnih, Volodymyr, Koray Kavukcuoglu, David Silver, Alex Graves, Ioannis Antonoglou, Daan Wierstra, and Martin Riedmiller. "Playing atari with deep reinforcement learning." arXiv preprint arXiv:1312.5602 (2013). Link.
  • Mnih, Volodymyr, Koray Kavukcuoglu, David Silver, Andrei A. Rusu, Joel Veness, Marc G. Bellemare, Alex Graves et al. "Human-level control through deep reinforcement learning." nature 518, no. 7540 (2015): 529-533. Link.