Google Git
Sign in
apache / submarine / 1c186a0a6ce9a08aeafc09042f70d3fe98b8f84b / . / dev-support / examples / mnist-tensorflow / MirroredStrategy
tree: 5fbc8f5a093de72655df884012f94c5f678b5c9d [path history] [tgz]
  1. build.sh
  2. Dockerfile
  3. mnist_keras_distributed.py
  4. post.sh
  5. readme.md
dev-support/examples/mnist-tensorflow/MirroredStrategy/readme.md

TF MirroredStrategy Example

Usage

This is an easy mnist example of how to train a distributed tensorflow model using MirroredStrategy and track the metric and paramater in submarine-sdk.

How to execute

  1. Set up (for a single terminal, only need to do this one time)
eval $(minikube -p minikube docker-env)
  1. Build the docker image
./dev-support/examples/mnist-tensorflow/MirroredStrategy/build.sh
  1. Submit a post request
./dev-support/examples/mnist-tensorflow/MirroredStrategy/post.sh