Skip to content

CartPole game by Reinforcement Learning, a journey from training to inference

License

Notifications You must be signed in to change notification settings

hypnosapos/cartpole-rl-remote

Repository files navigation

Cartpole RL Remote

Build Status License status

We love OpenSource

This project is intended to play with CartPole game using Reinforcement Learning and to know how we may train a different model experiments with enough observability (metrics/monitoring).

The model is divided basically in three parts: Neural network model, QLearning algorithm and application runner.

We want to show you a journey from custom training of models to a productive platform based on open source (training/inference).

Requirements

Basic scenario (Station #1):

  • Make (gcc)
  • Docker (18+)
  • Docker compose (version 17.06.+, compose file format 3.3)
  • python >= 3.5

Advanced scenarios (Station #2 and #3):

  • kubernetes (1.14+)
  • polyaxon (0.5.6)
  • seldon (0.4.0)

Station #1: Custom trainer and metrics collection

As with any other software development, machine learning code must follow the same best practices, so it's very important to have on mind that our code should be run on any environment, on my laptop or on any cloud (avoid vendor services, and ensuring portability anywhere).

First attempt was to train CartPole model with our own trainer by a multiprocessor python module, by default it'll try to use one processor for each hyperparameter combination (model experiment).

NOTE: Yes, we could have tried tensorboard callbacks for background Keras model (or tensorflow models), but we wanna explore other ways too.

Collecting metrics with visdom

We trust in logs, so all details of model training should be outlined using builtins log libraries, and then the instrumentation may come from tools that manage these log lines. We've used as first approach a log handler for Visdom server in order to send metrics to an external site.

Using python virtual env

Requirements:

  • Python (3.5+)

To create a local virtual env for python, type:

make venv

When this virtual env is activated, we can use the cartpole command client directly:

source .venv/bin/activate
cartpole --help

We have a couple of arguments to setup metrics collection: --metrics-engine and --metrics-config.

The simplest way to train the model and collect metrics with visdom (trough docker container) is next command

make train-dev

Change default values for hyperparameters in Makefile file if you wish another combination.

NOTE: Render mode is activated with -r argument if you want to see CartPole game training.

Visdom server might be ready at: http://localhost:8097 with metrics and evaluation model results, this process gets out an h5 file with the best trained model as well.

Using docker compose

If you prefer use docker containers for everything launch this command (powered by docker-compose):

make train-docker-visdom

Basic Scenario - Visdom

Using docker log drivers, EFK in action

Ok, it's possible to implement our own metrics collector, but as we are using containers, couldn't we use docker log drivers to extract metrics from log lines? Yes, of course.

We've created a fluentd conf file (under directory scaffold/efk/fluentd) to specify the regex pattern of searched lines in logs, and fluentd will send metrics to elasticsearch, finally visualizations of metrics will be available through kibana dashboard.

To run this stack type:

make train-docker-efk

Kibana URL would be: http://localhost:5601. Set the text cartpole-* as the index pattern. In scaffold/efk/kibana directory you can find a kibana dashboard json file that you can import to view all graphics about cartpole model experiments.

Basic Scenario - EFK

Using ModelDB as experiment repository

ModelDB is a MIT licensed project that let's track our model experiments pretty easy.

Check out typing:

make train-docker-modeldb

Frontend is available at: http://localhost:3000

Basic Scenario - ModelDB

Station #2: Advanced training with Polyaxon

Well, we have a simple model trainer with simple hyperparameter tuning implementation (something like a well known grid algorithm).

Few weeks ago I discovered polyaxon which goal is to train models seamlessly. The challenge now would be create a polyaxon wrapper to train multiple experiments.

It uses kubernetes as platform so first thing we need is create one cluster (take a look at k8s-gke):

export GCP_CREDENTIALS=/your_path/gcp.json
export GCP_ZONE=europe-west4-a
export GCP_PROJECT_ID=<my_project>
export GKE_CLUSTER_NAME=cartpole
export GITHUB_TOKEN=<githubtoken>
git clone https://github.com/hypnosapos/k8s-gke.git
make -C k8s-gke gke-bastion gke-create-cluster gke-tiller-helm gke-proxy gke-ui-login-skip gke-ui

We'll use a ZFS server to create shared volumes in the same GCP_ZONE (feel free to change de volume driver):

make -C scaffold/polyaxon gke-polyaxon-nfs

Install polyaxon components on kubernetes and configure the polyaxon client on gke-bastion container

make -C scaffold/polyaxon gke-polyaxon-preinstall gke-polyaxon-install gke-polyaxon-cartpole-init

Finally, let's deploy our experiment group by this command:

make  -C scaffold/polyaxon gke-polyaxon-cartpole

You can use the gke-bastion container as proxy for gcloud, kubectl or polyaxon commands directly, i.e:

docker exec -it gke-bastion sh -c "kubectl get pods -w -n polyaxon"
docker exec -it gke-bastion sh -c "polyaxon project experiments"

Here you have some screen shots of web console and command client

Polyaxon

Polyaxon Chart

Polyaxon Command Client

Station #3: Model inference with Seldon

The idea is to get trained models and deploy them within Seldon.

In order to create your own seldon images use:

make seldon-build seldon-push

This command uses the official seldon wrapper to build and push your docker images. Mainly the built image process attaches the best scored model (h5 file) to be served through the entry method "predict" for client requests when the seldon microservice is ready. Note that training models are moved from default ".models" local directory to scaffold/seldon directory to be included into the docker image, but obviously you can choose another, even from a cloud storage such as S3, GCS, ... (probably you are thinking about linking the output directory used in training stage with polyaxon, you're right).

We provide some docker images for this PoC with different scores under the dockerhub org hypnosapos.

Deploy Seldon

We're going to use the same kubernetes cluster, but you may to use another.

Deploy Seldon:

make gke-seldon-install

Deploy CartPole within Seldon

Deploy different seldon graphs for CartPole model, choose one value of: [model, abtest, router] for SELDON_MODEL_TYPE variable:

SELDON_MODEL_TYPE=router make gke-seldon-cartpole

Take a look at files under directory scaffold/k8s/seldon .

Let's deploy a router (it'll use an epsilon greedy router by seldon team) with three branches: two for "untrained" models ('cartpole-0' and 'cartpole-1', low score metric), and one branch with a "max_score" ('cartpole-2', score metric 7000, the max value in training). Default branch will be 0 ('cartpole-0') at the beginning, as requests are received the router will redirect traffic automatically to branch 2 ('cartpole-2') according to the best scored model.

Check out that pods are ready:

docker exec -it gke-bastion sh -c "kubectl get pods -l seldon-app=cartpole-router -w -n seldon"
NAME                                               READY     STATUS    RESTARTS   AGE
cartpole-router-cartpole-router-6678798bf4-4sz7x   5/5       Running   0          2m

docker exec -it gke-bastion sh -c 'kubectl get pods -l seldon-app=cartpole-router -o jsonpath="{.items[*].spec.containers[*].image}" -n seldon | tr -s "[[:space:]]" "\n"'
hypnosapos/cartpolerlremoteagent:untrained
hypnosapos/cartpolerlremoteagent:untrained
hypnosapos/cartpolerlremoteagent:max_score
seldonio/mab_epsilon_greedy:1.1
seldonio/engine:0.1.6

Run remote agent

You have to get the external IP from svc/seldon-apiserver to set RUN_MODEL_IP variable.

In order to get model predictions launch this command in your shell:

export RUN_MODEL_IP=$(docker exec -it gke-bastion sh -c \
'kubectl get svc seldon-apiserver -n seldon -o jsonpath="{.status.loadBalancer.ingress[0].ip}"')
make docker-visdom
make run-dev

Model metrics in running mode will be collected on local visdom server.

Take a look at the grafana dashboard to view seldon metrics. Since seldon-core-analytics helm chart was installed with loadbalancer endpoint type, find the public ip to get access.

Seldon router

Open-source platforms

We love OSS, we're K8S lovers and we think that this container orchestration engine is a key for the future of hybrid-clouds or multi-cloud strategies. Imagine you are capable to specify an AI pipeline (gathering ETL or another intensive tasks of pre-processing stages, model evaluation, serving, etc) via declarative or programmatic DSL

We recommend to take a look at one of the most widely contributed OSS project: kubeflow (we're focused on it today), some of the products shown above can be integrated within and you could be able to contribute or extend for your needs.

License

License Check

Authors