mirror of
https://github.com/wassname/ray.git
synced 2026-07-02 03:50:57 +08:00
Richard Liaw
36 lines
546 B
ReStructuredText
36 lines
546 B
ReStructuredText
Package Reference
|
|
=================
|
|
|
|
.. _ref-pytorch-trainer:
|
|
|
|
PyTorchTrainer
|
|
--------------
|
|
|
|
.. autoclass:: ray.util.sgd.pytorch.PyTorchTrainer
|
|
:members:
|
|
|
|
.. automethod:: __init__
|
|
|
|
.. _ref-pytorch-operator:
|
|
|
|
PyTorch TrainingOperator
|
|
------------------------
|
|
|
|
.. autoclass:: ray.util.sgd.pytorch.TrainingOperator
|
|
:members:
|
|
|
|
|
|
PyTorchTrainable
|
|
----------------
|
|
|
|
.. autoclass:: ray.util.sgd.pytorch.PyTorchTrainable
|
|
:members:
|
|
|
|
TFTrainer
|
|
---------
|
|
|
|
.. autoclass:: ray.util.sgd.tf.TFTrainer
|
|
:members:
|
|
|
|
.. automethod:: __init__
|