2020-01-29 08:51:01 -08:00
|
|
|
Package Reference
|
|
|
|
=================
|
|
|
|
|
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-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:
|
|
|
|
|
|
|
|
.. automethod:: __init__
|
|
|
|
|
2020-03-03 16:44:42 -08:00
|
|
|
.. _ref-torch-operator:
|
2020-03-01 21:22:48 -08:00
|
|
|
|
|
|
|
PyTorch TrainingOperator
|
|
|
|
------------------------
|
|
|
|
|
2020-03-03 16:44:42 -08:00
|
|
|
.. autoclass:: ray.util.sgd.torch.TrainingOperator
|
2020-03-01 21:22:48 -08:00
|
|
|
:members:
|
|
|
|
|
2020-01-29 08:51:01 -08:00
|
|
|
|
2020-03-03 16:44:42 -08:00
|
|
|
TorchTrainable
|
|
|
|
--------------
|
2020-01-29 08:51:01 -08:00
|
|
|
|
2020-03-03 16:44:42 -08:00
|
|
|
.. autoclass:: ray.util.sgd.torch.TorchTrainable
|
2020-01-29 08:51:01 -08:00
|
|
|
:members:
|
|
|
|
|
|
|
|
TFTrainer
|
|
|
|
---------
|
|
|
|
|
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__
|