Skip to content

GitLab

"vscode:/vscode.git/clone" did not exist on "bbb71033c680bdf8634f88cd16ff067184a478f4"
Commit 5b964dbb authored by A. Unique TensorFlower's avatar A. Unique TensorFlower
Browse files

Internal change 

PiperOrigin-RevId: 451473030
parent d1d418a5
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
official/common/distribute_utils.py
View file @ 5b964dbb
...@@ -96,7 +96,7 @@ def get_distribution_strategy(distribution_strategy="mirrored", ...@@ -96,7 +96,7 @@ def get_distribution_strategy(distribution_strategy="mirrored",
num_packs=1, num_packs=1,
tpu_address=None, tpu_address=None,
**kwargs): **kwargs):
"""Return a DistributionStrategy for running the model. """Return a Strategy for running the model.
Args: Args:
distribution_strategy: a string specifying which distribution strategy to distribution_strategy: a string specifying which distribution strategy to
...@@ -119,7 +119,7 @@ def get_distribution_strategy(distribution_strategy="mirrored", ...@@ -119,7 +119,7 @@ def get_distribution_strategy(distribution_strategy="mirrored",
**kwargs: Additional kwargs for internal usages. **kwargs: Additional kwargs for internal usages.
Returns: Returns:
tf.distribute.DistibutionStrategy object. tf.distribute.Strategy object.
Raises: Raises:
ValueError: if `distribution_strategy` is "off" or "one_device" and ValueError: if `distribution_strategy` is "off" or "one_device" and
`num_gpus` is larger than 1; or `num_gpus` is negative or if `num_gpus` is larger than 1; or `num_gpus` is negative or if
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment