tf.estimator.tpu.RunConfig

Если device_fn не None, он переопределяет функцию устройства по умолчанию, используемую в Estimator. В противном случае используется функция по умолчанию.

Все глобальные идентификаторы в кластере обучения назначаются из возрастающей последовательности последовательных целых чисел. Первый идентификатор равен 0.

Примечание: Идентификатор задачи (поле свойства task_id) отслеживает индекс узла среди всех узлов с тем же типом задачи. Например, при задании кластера следующим образом:

cluster = {'chief': ['host0:2222'],
'ps': ['host1:2222', 'host2:2222'],
'worker': ['host3:2222', 'host4:2222', 'host5:2222']}

Узлы с типом задачи worker могут иметь идентификаторы 0, 1, 2. Узлы с типом задачи ps могут иметь идентификаторы 0, 1. Следовательно, task_id не уникален, но пара (task_type, task_id) может однозначно определить узел в кластере.

Глобальный идентификатор, то есть это поле, отслеживает индекс узла среди ВСЕХ узлов в кластере. Он назначается уникально. Например, для указанного выше описания кластера глобальные идентификаторы назначаются следующим образом:

task_type  | task_id  |  global_id
--------------------------------
chief      | 0        |  0
worker     | 0        |  1
worker     | 1        |  2
worker     | 2        |  3
ps         | 0        |  4
ps         | 1        |  5