RaySGD API Documentation
========================

.. _ref-torch-trainer:

TorchTrainer
------------

.. autoclass:: ray.util.sgd.torch.TorchTrainer
    :members:

    .. automethod:: __init__

.. _ref-torch-operator:

PyTorch TrainingOperator
------------------------

.. autoclass:: ray.util.sgd.torch.TrainingOperator
    :members:

.. _BaseTorchTrainable-doc:

BaseTorchTrainable
------------------

.. autoclass:: ray.util.sgd.torch.BaseTorchTrainable
    :members:
    :private-members:

TFTrainer
---------

.. autoclass:: ray.util.sgd.tf.TFTrainer
    :members:

    .. automethod:: __init__