[tune] TensorFlow Distributed Trainable (#11876)

Co-authored-by: Richard Liaw <rliaw@berkeley.edu>
This commit is contained in:
Keqiu Hu
2020-11-10 14:59:08 -08:00
committed by GitHub
co-authored by Richard Liaw
parent 50dbf1a307
commit 0c1bdaef59
6 changed files with 395 additions and 1 deletions
+12 -1
View File
@@ -352,7 +352,7 @@ Utilities
Distributed Torch
-----------------
Ray also offers lightweight integrations to distribute your model training on Ray Tune.
Ray offers lightweight integrations to distribute your PyTorch training on Ray Tune.
.. autofunction:: ray.tune.integration.torch.DistributedTrainableCreator
@@ -364,6 +364,17 @@ Ray also offers lightweight integrations to distribute your model training on Ra
.. autofunction:: ray.tune.integration.torch.is_distributed_trainable
:noindex:
.. _tune-dist-tf-doc:
Distributed TensorFlow
----------------------
Ray also offers lightweight integrations to distribute your TensorFlow training on Ray Tune.
.. autofunction:: ray.tune.integration.tensorflow.DistributedTrainableCreator
:noindex:
tune.DurableTrainable
---------------------