/rl-pendulum

Open AI Gym - Pendulum-v1 reinforcement learning (DQN, SAC)

Primary LanguagePythonMIT LicenseMIT

Pendulum Reinforcement Learning

Open AI Gym - Pendulum-v1

This repository implements the pendulum example through applying DQN and SAC algorithm.

Requirements

  • pytorch
  • gym

Train

  • Train through SAC method (you can change the name of log files folder at "line 216")
python3 Pendulum_SAC/pendulum_sac.py
  • Train through DQN method (you can change the name of log files folder at "line 132")
python3 Pendulum_DQN/pendulum_dqn.py

Results