/pytorch-REINFORCE

PyTorch Implementation of REINFORCE for both discrete & continuous control

Primary LanguagePython

PyTorch REINFORCE

PyTorch implementation of REINFORCE.
This repo supports both continuous and discrete environments in OpenAI gym.

Requirement

  • python 2.7
  • PyTorch
  • OpenAI gym
  • Mujoco (optional)

Run

Use the default hyperparameters. (Program will detect whether the environment is continuous or discrete)

python main.py --env_name [name of environment]

Experiment results

continuous: InvertedPendulum-v1

discrete: CartPole-v0

Reference