tf.distribute.cluster_resolver.SimpleClusterResolver
| Просмотреть исходный код на GitHub |
Простое реализация ClusterResolver, принимающая ClusterSpec.
Наследуется от: ClusterResolver
tf.distribute.cluster_resolver.SimpleClusterResolver(
cluster_spec, master='', task_type=None, task_id=None, environment='',
num_accelerators=None, rpc_layer=None
)
| Атрибуты | |
|---|---|
environment | Возвращает текущую среду, в которой выполняется TensorFlow. Возможны два значения возврата: «google» (когда TensorFlow выполняется во внутренней среде Google) или пустая строка (когда TensorFlow выполняется где-либо еще). Если вы реализуете ClusterResolver, который работает как в среде Google, так и в открытом мире (например, TPU ClusterResolver или аналогичный), вам необходимо вернуть соответствующую строку в зависимости от среды, которую вам нужно определить. В противном случае, если вы реализуете ClusterResolver, который будет работать только в открытом TensorFlow, вам не нужно реализовывать это свойство. |
rpc_layer | |
task_id | |
task_type | |
Методы
cluster_spec
cluster_spec()
Возвращает ClusterSpec, переданный в конструктор.
master
master(
task_type=None, task_id=None, rpc_layer=None
)
Возвращает адрес узла для использования при создании сессии.
| Аргументы | |
|---|---|
task_type | (Необязательно) Тип задачи TensorFlow узла. |
task_id | (Необязательно) Индекс задачи TensorFlow узла. |
rpc_layer | (Необязательно) RPC, используемый распределенным TensorFlow. |
| Возвращаемые значения | |
|---|---|
| Имя или URL узла сессии. |
Если заданы task_type и task_id, это переопределит строку master , переданную в функцию инициализации.
num_accelerators
num_accelerators(
task_type=None, task_id=None, config_proto=None
)
Возвращает количество ускорительных ядер на рабочий узел.
SimpleClusterResolver не выполняет автоматическое обнаружение ускорителей, поэтому сессия TensorFlow никогда не будет создана, и все аргументы не используются, и мы просто предполагаем, что тип ускорителя — GPU, и возвращаем значение, переданное нам в конструкторе.
| Аргументы | |
|---|---|
task_type | Не используется. |
task_id | Не используется. |
config_proto | Не используется. |
© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r1.15/api_docs/python/tf/distribute/cluster_resolver/SimpleClusterResolver