/SimplePytorchActorCritic

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

Primary LanguagePythonMIT LicenseMIT

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