mirror of
https://github.com/vale981/ray
synced 2025-03-06 18:41:40 -05:00
3a134c7224
* Start renaming pytorch to torch * Rename PyTorchTrainer to TorchTrainer * Rename PyTorch runners to Torch runners * Finish renaming API * Rename to torch in tests * Finish renaming docs + tests * Run format + fix DeprecationWarning * fix * move tests up * rename Co-authored-by: Richard Liaw <rliaw@berkeley.edu>
35 lines
524 B
ReStructuredText
35 lines
524 B
ReStructuredText
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__
|