Skip to content

Latest commit

 

History

History
33 lines (21 loc) · 745 Bytes

README.md

File metadata and controls

33 lines (21 loc) · 745 Bytes

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