Package Reference
=================

.. _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:


TorchTrainable
--------------

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

TFTrainer
---------

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

    .. automethod:: __init__