Back to Ray

MNIST PyTorch Example

doc/source/tune/examples/includes/mnist_pytorch.rst

1.13.1366 B
Original Source

:orphan:

MNIST PyTorch Example


.. literalinclude:: /../../python/ray/tune/examples/mnist_pytorch.py

If you consider switching to PyTorch Lightning to get rid of some of your boilerplate
training code, please know that we also have a walkthrough on :doc:`how to use Tune with
PyTorch Lightning models </tune/examples/tune-pytorch-lightning>`.