Modify logging which is specific to ParameterServerStrategy.
PiperOrigin-RevId: 278431437
Change-Id: I7b3cd16954c9f2830e6e57669bc6dd3f7cec988b
diff --git a/tensorflow/python/distribute/parameter_server_strategy.py b/tensorflow/python/distribute/parameter_server_strategy.py
index 2d3800e..1815fc2 100644
--- a/tensorflow/python/distribute/parameter_server_strategy.py
+++ b/tensorflow/python/distribute/parameter_server_strategy.py
@@ -299,8 +299,9 @@
self._task_id = None
logging.info(
- "ParameterServerStrategy with compute_devices = %r, "
- "variable_device = %r", compute_devices, self._variable_device)
+ "ParameterServerStrategy (CentralStorageStrategy if you are using a "
+ "single machine) with compute_devices = %r, variable_device = %r",
+ compute_devices, self._variable_device)
def _validate_colocate_with_variable(self, colocate_with_variable):
values.validate_colocate(colocate_with_variable, self)