Skip to content

Minimal implementations of RL algorithms using PyTorch

License

Notifications You must be signed in to change notification settings

r-salas/minimal-rl

Repository files navigation

minimal-rl

RL algorithms with minimal lines of code

Installation

$ git clone https://github.com/r-salas/minimal-rl.git
$ cd minimal-rl
$ pip install -r requirements.txt

If you're in Linux, you may also need to install swig:

$ apt-get install swig

DQN

$ python dqn.py

Solving LunarLander-v2 with DQN

DDPG

$ python ddpg.py

Solving MountainCarContinous-v0 with DDPG

REINFORCE

$ python reinforce.py

Solving CartPole-v1 with REINFORCE

A2C

$ python a2c.py

Solving Pendulum-v0 with A2C

CEM

$ python cem.py

Solving Acrobot-v1 with CEM