Google Git
Sign in
apache / submarine / refs/heads/SUBMARINE-1393 / . / dev-support / examples / mnist-tensorflow / MirroredStrategy
tree: d3ef7853f23d02ab231085d40c1341f92b06803a [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