Skip to content

Latest commit

 

History

History
20 lines (13 loc) · 1.16 KB

File metadata and controls

20 lines (13 loc) · 1.16 KB

Simple Cartpole Example

What is cartpole?

This is a classic example of a problem that reinforcement learning can solve. It's a simulation of balancing a broom upright by balancing it on your hand. The broom is the "pole" and your hand is replaced with a "cart" moving back and forth on a linear track. This simplified example works in 2 dimensions, so the cart can only move in a line back and forth, and the pole can only fall forwards or backwards, not to the sides.

This Example

Sample Cart Pole Example using SageMaker RL with base docker image containing Coach, MxNet and OpenAI Gym. This is a toy example taken from Coach Quick Start Guide. It demonstrates how you can use the RLEstimator from the SageMaker Python SDK in script mode.

Summary

The sample notebook demonstrates how to:

  1. Train a toy cart pole model from a notebook and python SDK/script mode
  2. Visualize gifs generated during training.
  3. Move these back and forth from S3 (while running in SageMaker mode)
  4. Checkpoint trained model
  5. Run inference using checkpointed models