def __init__(self, compute_devices=None, parameter_device=None):
"""Initializes this strategy with default TFConfigClusterResolver."""
super(CentralStorageStrategyV1, self).__init__(
parameter_server_strategy.ParameterServerStrategyExtended(
self,
compute_devices=compute_devices,
parameter_device=parameter_device))
def __init__(self, compute_devices=None, parameter_device=None):
super(CentralStorageStrategyV1, self).__init__(
parameter_server_strategy.ParameterServerStrategyExtended(
self,
compute_devices=compute_devices,
parameter_device=parameter_device))
distribute_lib.distribution_strategy_gauge.get_cell('V1').set(
'CentralStorageStrategy')
def __init__(self, cluster_resolver=None):
"""Initializes this strategy."""
# The `cluster_resolver` must be set so that
# `ParameterServerStrategyExtended` will keep num_gpus for `configure`
# method.
if cluster_resolver is None:
cluster_resolver = TFConfigClusterResolver()
extended = parameter_server_strategy.ParameterServerStrategyExtended(
self, cluster_resolver=cluster_resolver)
super(ParameterServerStrategy, self).__init__(extended)
def __init__(self, compute_devices=None, parameter_device=None):
extended = parameter_server_strategy.ParameterServerStrategyExtended(
self,
compute_devices=compute_devices,
parameter_device=parameter_device)
"""Initializes the strategy with optional device strings.
Args:
compute_devices: an optional list of strings for device to replicate models
on. If this is not provided, all local GPUs will be used; if there is no
GPU, local CPU will be used.
parameter_device: an optional device string for which device to put
variables on. The default one is CPU or GPU if there is only one.
"""
super(CentralStorageStrategy, self).__init__(extended)
def __init__(self, cluster_resolver):
super(MockCoreParameterServerStrategy, self).__init__(
core_parameter_server_strategy.ParameterServerStrategyExtended(
self, cluster_resolver=cluster_resolver))
def __init__(self, compute_devices=None, parameter_device=None):
extended = parameter_server_strategy.ParameterServerStrategyExtended(
self,
compute_devices=compute_devices,
parameter_device=parameter_device)
super(CentralStorageStrategy, self).__init__(extended)
