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

.. _ref-pytorch-trainer:

PyTorchTrainer
--------------

.. autoclass:: ray.experimental.sgd.pytorch.PyTorchTrainer
    :members:

    .. automethod:: __init__


PyTorchTrainable
----------------

.. autoclass:: ray.experimental.sgd.pytorch.PyTorchTrainable
    :members:

TFTrainer
---------

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

    .. automethod:: __init__