/pytorch-REINFORCE

PyTorch Implementation of REINFORCE for both discrete & continuous control

Primary LanguagePython

PyTorch REINFORCE

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

Requirement

  • python 2.7
  • PyTorch
  • OpenAI gym
  • Roboschool

Run

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

Experiment results

continuous: InvertedPendulum-v1

Reference