Add document for Orca PytorchRayEstimator and PytorchSparkEstimator (#3458)

* add document for pytorch_ray_estimator and pytorch_spark_estimator

* change module to class

* fix typo
This commit is contained in:
Yu Shan 2021-11-11 19:41:37 +08:00 committed by GitHub
parent dffc40630b
commit 829ee31f1b

View file

@ -35,6 +35,29 @@ orca.learn.pytorch.estimator
:undoc-members:
:show-inheritance:
orca.learn.pytorch.pytorch_ray_estimator
^^^^^^^^^^^^^^^^^^^^^^^^^^^
Orca Pytorch Estimator with backend of "horovod" or "torch_distributed".
.. autoclass:: bigdl.orca.learn.pytorch.pytorch_ray_estimator.PyTorchRayEstimator
:members:
:undoc-members:
:show-inheritance:
orca.learn.pytorch.pytorch_spark_estimator
^^^^^^^^^^^^^^^^^^^^^^^^^^^
Orca Pytorch Estimator with backend of "bigdl".
.. autoclass:: bigdl.orca.learn.pytorch.pytorch_spark_estimator.PyTorchSparkEstimator
:members:
:undoc-members:
:show-inheritance:
orca.learn.openvino.estimator
------------------------------