Rename Orca Tensorflow Estimator backend name from "tf2" to "ray" (#4938)
* rename backend name from tf2 to ray * rename in doc * rename in docstring * manually * rename test_tf2estimator_tf2_backend.py -> test_tf2estimator_ray_backend.py * change tf2 in multi_agent_two_trainers.py back
This commit is contained in:
parent
8f51ed4cf1
commit
86fa1c02ab
2 changed files with 3 additions and 3 deletions
|
|
@ -64,10 +64,10 @@ View the related [Python API doc](https://bigdl.readthedocs.io/en/latest/doc/Pyt
|
||||||
|
|
||||||
#### **2.2 TensorFlow 2.x and Keras 2.4+**
|
#### **2.2 TensorFlow 2.x and Keras 2.4+**
|
||||||
|
|
||||||
**Using `tf2` or *Horovod* backend**
|
**Using `ray` or *Horovod* backend**
|
||||||
|
|
||||||
Users can create an `Estimator` for TensorFlow 2.x from a Keras model (using a _Model Creator Function_) when the backend is
|
Users can create an `Estimator` for TensorFlow 2.x from a Keras model (using a _Model Creator Function_) when the backend is
|
||||||
`tf2` (currently default for TF2) or *Horovod*. For example:
|
`ray` (currently default for TF2) or *Horovod*. For example:
|
||||||
|
|
||||||
```python
|
```python
|
||||||
def model_creator(config):
|
def model_creator(config):
|
||||||
|
|
|
||||||
|
|
@ -31,7 +31,7 @@ orca.learn.tf2.estimator
|
||||||
orca.learn.tf2.tf2_ray_estimator
|
orca.learn.tf2.tf2_ray_estimator
|
||||||
^^^^^^^^^^^^^^^^^^^^^^^^^^^
|
^^^^^^^^^^^^^^^^^^^^^^^^^^^
|
||||||
|
|
||||||
Orca TF2Estimator with backend of "horovod" or "tf2".
|
Orca TF2Estimator with backend of "horovod" or "ray".
|
||||||
|
|
||||||
.. autoclass:: bigdl.orca.learn.tf2.ray_estimator.TensorFlow2Estimator
|
.. autoclass:: bigdl.orca.learn.tf2.ray_estimator.TensorFlow2Estimator
|
||||||
:members:
|
:members:
|
||||||
|
|
|
||||||
Loading…
Reference in a new issue