Skip to content

Latest commit

 

History

History
41 lines (32 loc) · 796 Bytes

README.md

File metadata and controls

41 lines (32 loc) · 796 Bytes

DRL-pytorch

DQN, DDPG, PPO, SAC...in pytorch for gym envs

Installation

Clone this repo:

  git clone https://github.com/ZDDWLIG/DRL-pytorch.git
  cd path/to/DRL-pytorch-master

Use conda to manage the python environment:

  conda create -n DRL-pytorch python=3.8
  conda activate DRL-pytorch
  pip install -r requirements.txt

Train

For example, if you want to train DQN, then run:

  cd DQN
  python train.py

The checkpoints and logs can be found in results

Test

If you want to test DQN, then run:

  cd DQN
  python test.py

Demo

LunarLander-v2

LunarLander-v2 GIF

Pendulum-v1

Pendulum-v1 GIF