mirror of
https://github.com/wassname/ray.git
synced 2026-06-29 03:37:20 +08:00
Amog Kamsetty
58 lines
839 B
ReStructuredText
58 lines
839 B
ReStructuredText
RaySGD API Documentation
|
|
========================
|
|
|
|
.. _ref-torch-trainer:
|
|
|
|
TorchTrainer
|
|
------------
|
|
|
|
.. autoclass:: ray.util.sgd.torch.TorchTrainer
|
|
:members:
|
|
|
|
.. _ref-torch-operator:
|
|
|
|
PyTorch TrainingOperator
|
|
------------------------
|
|
|
|
.. autoclass:: ray.util.sgd.torch.TrainingOperator
|
|
:members:
|
|
|
|
.. _BaseTorchTrainable-doc:
|
|
|
|
BaseTorchTrainable
|
|
------------------
|
|
|
|
.. autoclass:: ray.util.sgd.torch.BaseTorchTrainable
|
|
:members:
|
|
:private-members:
|
|
|
|
TFTrainer
|
|
---------
|
|
|
|
.. autoclass:: ray.util.sgd.tf.TFTrainer
|
|
:members:
|
|
|
|
.. automethod:: __init__
|
|
|
|
Dataset
|
|
-------
|
|
|
|
.. autoclass:: ray.util.sgd.data.Dataset
|
|
:members:
|
|
|
|
.. automethod:: __init__
|
|
|
|
.. _ref-utils:
|
|
|
|
Utils
|
|
-----
|
|
|
|
.. autoclass:: ray.util.sgd.utils.AverageMeter
|
|
:members:
|
|
|
|
.. autoclass:: ray.util.sgd.utils.AverageMeterCollection
|
|
:members:
|
|
|
|
|
|
|