qingshi9974 / DQN-pytorch-Atari

Implement DQN and DDQN algorithm on Atari games,such as BreakoutNoFrameskip-v4, PongNoFrameskip-v4,BoxingNoFrameskip-v4.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

DQN-pytorch-Atari

Implement DQN and DDQN algorithm on Atari games,such as BreakoutNoFrameskip-v4, PongNoFrameskip-v4,BoxingNoFrameskip-v4.

Requirements

  • cuda 10.1
  • python 3.7.6
  • gym 0.17.6
  • pytorch 1.5.0
  • tdqm

Usage Example

$ python main.py --env_name BreakoutNoFrameskip-v4-v2

Results without tricks in OpenAI Baselines

BreakoutNoFrameskip-v4

imageimageimage

Results with tricks in OpenAI Baselines

BreakoutNoFrameskip-v4

image

PongNoFrameskip-v4

image

BoxingNoFrameskip-v4

image

About

Implement DQN and DDQN algorithm on Atari games,such as BreakoutNoFrameskip-v4, PongNoFrameskip-v4,BoxingNoFrameskip-v4.


Languages

Language:Python 100.0%