ray/doc/source/raysgd/raysgd_ref.rst

36 lines
524 B
ReStructuredText
Raw Normal View History

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__