From 7364a0eb6c55217eddba1fb4f44cd0f727cba96e Mon Sep 17 00:00:00 2001 From: Kanchi Shimono Date: Sun, 20 Jun 2021 22:57:37 +0900 Subject: [PATCH] Fix docstrings of MirroredStrategyRunner --- .../mirrored_strategy_runner.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/spark/spark-tensorflow-distributor/spark_tensorflow_distributor/mirrored_strategy_runner.py b/spark/spark-tensorflow-distributor/spark_tensorflow_distributor/mirrored_strategy_runner.py index 661b9641..fd0db987 100644 --- a/spark/spark-tensorflow-distributor/spark_tensorflow_distributor/mirrored_strategy_runner.py +++ b/spark/spark-tensorflow-distributor/spark_tensorflow_distributor/mirrored_strategy_runner.py @@ -89,10 +89,10 @@ def __init__(self, a discovery script that is formatted according to the Spark configuration docs. Make sure `spark.driver.resource.{gpu_resource_name}.discoveryScript` and - `spark.driver.resource.{gpu_resource_name}.discoveryScript` are - also set in the Spark conf. In particular, the GPU addresses - should be zero indexed. For example, the output of the - discovery script for 3 GPUs with gpu_resource_name='gpu' + `spark.executor.resource.{gpu_resource_name}.discoveryScript` + are also set in the Spark conf. In particular, the GPU + addresses should be zero indexed. For example, the output of + the discovery script for 3 GPUs with gpu_resource_name='gpu' would be `{"name": "gpu", "addresses":["0","1","2"]}`. See an example discovery script: `github.com/apache/spark/blob/ master/examples/src/main/scripts/getGpusResources.sh`.