Skip to content

chingyaoc/pytorch-REINFORCE

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

13 Commits
 
 
 
 
 
 
 
 
 
 
 
 

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

 
 
 

Contributors

Languages