mirror of
https://github.com/vale981/ray
synced 2025-03-07 02:51:39 -05:00
27 lines
400 B
ReStructuredText
27 lines
400 B
ReStructuredText
Package Reference
|
|
=================
|
|
|
|
.. _ref-pytorch-trainer:
|
|
|
|
PyTorchTrainer
|
|
--------------
|
|
|
|
.. autoclass:: ray.util.sgd.pytorch.PyTorchTrainer
|
|
:members:
|
|
|
|
.. automethod:: __init__
|
|
|
|
|
|
PyTorchTrainable
|
|
----------------
|
|
|
|
.. autoclass:: ray.util.sgd.pytorch.PyTorchTrainable
|
|
:members:
|
|
|
|
TFTrainer
|
|
---------
|
|
|
|
.. autoclass:: ray.util.sgd.tf.TFTrainer
|
|
:members:
|
|
|
|
.. automethod:: __init__
|