Skip to content

PyTorch Implementation of REINFORCE for both discrete & continuous control

Notifications You must be signed in to change notification settings

chingyaoc/pytorch-REINFORCE

Repository files navigation

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

About

PyTorch Implementation of REINFORCE for both discrete & continuous control

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages