https://github.com/chingyaoc/pytorch-REINFORCE

PyTorch Implementation of REINFORCE for both discrete & continuous control
https://github.com/chingyaoc/pytorch-REINFORCE

continuous-control gym mujoco pytorch reinforce reinforcement-learning

Last synced: about 2 months ago
JSON representation

PyTorch Implementation of REINFORCE for both discrete & continuous control

• Host: GitHub
• URL: https://github.com/chingyaoc/pytorch-REINFORCE
• Owner: chingyaoc
• Created: 2017-04-16T04:13:08.000Z (over 7 years ago)
• Default Branch: master
• Last Pushed: 2017-04-16T16:28:38.000Z (over 7 years ago)
• Last Synced: 2024-05-23T06:49:34.042Z (4 months ago)
• Topics: continuous-control, gym, mujoco, pytorch, reinforce, reinforcement-learning
• Language: Python
• Homepage:
• Size: 330 KB
• Stars: 261
• Watchers: 13
• Forks: 51
• Open Issues: 5
• Metadata Files:
  • Readme: README.md

Awesome Lists containing this project

• Awesome-pytorch-list-CNVersion - pytorch-REINFORCE

README

        
# 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

- [pytorch example](https://github.com/pytorch/examples/blob/master/reinforcement_learning/reinforce.py)