mirror of
https://github.com/vale981/ray
synced 2025-03-09 12:56:46 -04:00
72cc0c9bda
* [SGDv2] Add Tune-Cifar-PyTorch-PBT example * Update python/ray/util/sgd/v2/BUILD * Lint * Update example * Update docs
79 lines
1.8 KiB
ReStructuredText
79 lines
1.8 KiB
ReStructuredText
.. _sgd-v2-examples:
|
|
|
|
RaySGD Examples
|
|
===============
|
|
|
|
.. Example .rst files should be organized in the same manner as the
|
|
.py files in ray/python/ray/util/sgd/v2/examples.
|
|
|
|
Below are examples for using RaySGD with a variety of models, frameworks, and use cases.
|
|
|
|
General Examples
|
|
----------------
|
|
|
|
PyTorch
|
|
~~~~~~~
|
|
|
|
* :doc:`/raysgd/v2/examples/train_linear_example`:
|
|
Simple example for PyTorch.
|
|
|
|
* :doc:`/raysgd/v2/examples/train_fashion_mnist_example`:
|
|
End-to-end example for PyTorch.
|
|
|
|
* :doc:`/raysgd/v2/examples/transformers/transformers_example`:
|
|
End-to-end example for HuggingFace Transformers (PyTorch).
|
|
|
|
TensorFlow
|
|
~~~~~~~~~~
|
|
|
|
* :doc:`/raysgd/v2/examples/tensorflow_mnist_example`:
|
|
End-to-end example for TensorFlow
|
|
|
|
Horovod
|
|
~~~~~~~~~~
|
|
|
|
* :doc:`/raysgd/v2/examples/horovod/horovod_example`:
|
|
End-to-end example for Horovod (with PyTorch)
|
|
|
|
|
|
..
|
|
TODO
|
|
|
|
* :doc:`/raysgd/v2/examples/TODO`:
|
|
Simple example for TensorFlow
|
|
|
|
* :doc:`/raysgd/v2/examples/TODO`:
|
|
Simple example for Horovod (with TensorFlow)
|
|
|
|
|
|
Iterator API Examples
|
|
---------------------
|
|
|
|
* :doc:`/raysgd/v2/examples/mlflow_fashion_mnist_example`:
|
|
Example for using the Iterator API for custom MLFlow integration.
|
|
|
|
Ray Tune Integration Examples
|
|
-----------------------------
|
|
|
|
* :doc:`/raysgd/v2/examples/tune_linear_example`:
|
|
Simple example for tuning a PyTorch model.
|
|
|
|
* :doc:`/raysgd/v2/examples/tune_tensorflow_mnist_example`:
|
|
End-to-end example for tuning a TensorFlow model.
|
|
|
|
* :doc:`/raysgd/v2/examples/tune_cifar_pytorch_pbt_example`:
|
|
End-to-end example for tuning a PyTorch model with PBT.
|
|
|
|
..
|
|
TODO implement these examples!
|
|
|
|
Features
|
|
--------
|
|
|
|
* Example for using a custom callback
|
|
* End-to-end example for running on an elastic cluster (elastic training)
|
|
|
|
Models
|
|
------
|
|
|
|
* Example training on Vision model.
|