2020-11-28 10:09:38 -08:00
|
|
|
RaySGD API Reference
|
|
|
|
====================
|
|
|
|
|
|
|
|
PyTorch
|
|
|
|
-------
|
2020-01-29 08:51:01 -08:00
|
|
|
|
2020-03-03 16:44:42 -08:00
|
|
|
.. _ref-torch-trainer:
|
2020-01-29 08:51:01 -08:00
|
|
|
|
2020-03-03 16:44:42 -08:00
|
|
|
TorchTrainer
|
2020-11-28 10:09:38 -08:00
|
|
|
~~~~~~~~~~~~
|
2020-01-29 08:51:01 -08:00
|
|
|
|
2020-03-03 16:44:42 -08:00
|
|
|
.. autoclass:: ray.util.sgd.torch.TorchTrainer
|
2020-01-29 08:51:01 -08:00
|
|
|
:members:
|
|
|
|
|
2020-03-03 16:44:42 -08:00
|
|
|
.. _ref-torch-operator:
|
2020-03-01 21:22:48 -08:00
|
|
|
|
|
|
|
PyTorch TrainingOperator
|
2020-11-28 10:09:38 -08:00
|
|
|
~~~~~~~~~~~~~~~~~~~~~~~~
|
2020-03-01 21:22:48 -08:00
|
|
|
|
2020-03-03 16:44:42 -08:00
|
|
|
.. autoclass:: ray.util.sgd.torch.TrainingOperator
|
2020-03-01 21:22:48 -08:00
|
|
|
:members:
|
|
|
|
|
2020-11-28 10:09:38 -08:00
|
|
|
.. _ref-creator-operator:
|
|
|
|
|
|
|
|
CreatorOperator
|
|
|
|
~~~~~~~~~~~~~~~~
|
|
|
|
|
|
|
|
.. autoclass:: ray.util.sgd.torch.training_operator.CreatorOperator
|
|
|
|
:members:
|
|
|
|
:exclude-members: setup
|
|
|
|
|
|
|
|
.. _ref-lightning-operator:
|
|
|
|
|
|
|
|
Pytorch Lightning LightningOperator
|
|
|
|
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
|
|
|
|
|
|
|
|
.. autoclass:: ray.util.sgd.torch.lightning_operator.LightningOperator
|
|
|
|
:members:
|
|
|
|
:exclude-members: setup, train_epoch, train_batch, validate, validate_batch, state_dict, load_state_dict
|
|
|
|
|
2020-03-30 10:58:49 -07:00
|
|
|
.. _BaseTorchTrainable-doc:
|
2020-01-29 08:51:01 -08:00
|
|
|
|
2020-03-30 10:58:49 -07:00
|
|
|
BaseTorchTrainable
|
2020-11-28 10:09:38 -08:00
|
|
|
~~~~~~~~~~~~~~~~~~
|
2020-01-29 08:51:01 -08:00
|
|
|
|
2020-03-30 10:58:49 -07:00
|
|
|
.. autoclass:: ray.util.sgd.torch.BaseTorchTrainable
|
2020-01-29 08:51:01 -08:00
|
|
|
:members:
|
2020-03-30 10:58:49 -07:00
|
|
|
:private-members:
|
2020-01-29 08:51:01 -08:00
|
|
|
|
2020-11-28 10:09:38 -08:00
|
|
|
Tensorflow
|
|
|
|
----------
|
|
|
|
|
2020-01-29 08:51:01 -08:00
|
|
|
TFTrainer
|
2020-11-28 10:09:38 -08:00
|
|
|
~~~~~~~~~
|
2020-01-29 08:51:01 -08:00
|
|
|
|
2020-02-18 13:43:19 -08:00
|
|
|
.. autoclass:: ray.util.sgd.tf.TFTrainer
|
2020-01-29 08:51:01 -08:00
|
|
|
:members:
|
|
|
|
|
|
|
|
.. automethod:: __init__
|
2020-06-01 15:48:15 -07:00
|
|
|
|
2020-11-28 10:09:38 -08:00
|
|
|
RaySGD Dataset
|
|
|
|
---------------
|
|
|
|
|
2020-06-01 15:48:15 -07:00
|
|
|
Dataset
|
2020-11-28 10:09:38 -08:00
|
|
|
~~~~~~~
|
2020-06-01 15:48:15 -07:00
|
|
|
|
|
|
|
.. autoclass:: ray.util.sgd.data.Dataset
|
|
|
|
:members:
|
|
|
|
|
|
|
|
.. automethod:: __init__
|
|
|
|
|
2020-11-28 10:09:38 -08:00
|
|
|
RaySGD Utils
|
|
|
|
-------------
|
2020-08-19 17:22:28 -07:00
|
|
|
.. _ref-utils:
|
|
|
|
|
|
|
|
Utils
|
2020-11-28 10:09:38 -08:00
|
|
|
~~~~~
|
2020-08-19 17:22:28 -07:00
|
|
|
|
|
|
|
.. autoclass:: ray.util.sgd.utils.AverageMeter
|
|
|
|
:members:
|
|
|
|
|
|
|
|
.. autoclass:: ray.util.sgd.utils.AverageMeterCollection
|
|
|
|
:members:
|
|
|
|
|
|
|
|
|
|
|
|
|