Files
ray/doc/source/raysgd/raysgd_ref.rst
T
Richard Liaw 037aa2b961 [sgd] Refactor PyTorch SGD Documentation. (#6910)
* Refactor documentation and directory structurre

* update loss

* ,ore examples

* fix comments

* more code

* svgs

* formatting

* more_docs

* more writing

* comments ready

* move

* whitespace

* examples

* fix

* bold

* pytorch

* batch

* fix

* fix test

* Apply suggestions from code review

* quarantinegp

* tests/

* fix missing
2020-01-29 08:51:01 -08:00

28 lines
424 B
ReStructuredText

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__