Back to Ray

Training with TensorFlow and Ray Train

doc/source/train/examples/tf/tensorflow_mnist_example.rst

1.13.1804 B
Original Source

:orphan:

Training with TensorFlow and Ray Train

.. raw:: html

<a id="try-anyscale-quickstart-tensorflow_mnist_example" target="_blank" href="https://console.anyscale.com/register/ha?render_flow=ray&utm_source=ray_docs&utm_medium=docs&utm_campaign=tensorflow_mnist_example">
  
  


</a>

This basic example runs distributed training of a TensorFlow model on MNIST with Ray Train.

Code example

.. literalinclude:: /../../python/ray/train/examples/tf/tensorflow_mnist_example.py

See also

  • :doc:Ray Train Examples <../../examples> for more use cases.

  • :ref:Distributed Tensorflow & Keras <train-tensorflow-overview> for a tutorial.