Skip to content

Caffe2 Reinforcement Learning via Gradient Descent on Cartpole

Notifications You must be signed in to change notification settings

keithmgould/caffe2_cartpole_reinforce

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

7 Commits
 
 
 
 
 
 

Repository files navigation

Intro

Uses Caffe2 to implement reinforcement learning via gradient descent.

The problem solved is the openAI Gym Cartpole problem.

Status

Currently the script manages the forward pass, but I don't get how to do the backward pass within the context of Caffe2.

Inspiration

This script is inspired by this Pytorch example

About

Caffe2 Reinforcement Learning via Gradient Descent on Cartpole

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

 
 
 

Languages