gillesdami / SimplePytorchActorCritic

Pytorch actor critic model capable of solving 'CartPole-v0' with tensorboard

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Simple Pytorch Actor Critic

This repository feature a minimalist actor critic model capable of solving 'CartPole-v0' gym environment. The model is written in pytorch and uses tensorboardX.

tensorboard_screen.png

Train & Run

This project require python3.5+ torch, gym and tensorboardX

python main.py #train the model
python run.py #display an episode

About

Pytorch actor critic model capable of solving 'CartPole-v0' with tensorboard

License:MIT License


Languages

Language:Python 100.0%