Python tf.config.experimental.ClusterDeviceFilters用法及代碼示例

表示集群中遠程工作人員的設備過濾器集合。

用法

tf.config.experimental.ClusterDeviceFilters()

注意：這是一個實驗性 API，可能會發生變化。

為選擇性作業和任務設置設備過濾器。對於每個遠程工作人員，設備過濾器是一個字符串列表。當存在任何過濾器時，遠程工作人員將忽略與其任何過濾器不匹配的所有設備。每個過濾器都可以部分指定，例如"/job:ps"、"/job:worker/replica:3" 等。請注意，設備對其所在的工作人員始終可見。

例如，為參數服務器集群設置設備過濾器：

cdf = tf.config.experimental.ClusterDeviceFilters()
for i in range(num_workers):
  cdf.set_device_filters('worker', i, ['/job:ps'])
for i in range(num_ps):
  cdf.set_device_filters('ps', i, ['/job:worker'])

tf.config.experimental_connect_to_cluster(cluster_def,
                                          cluster_device_filters=cdf)

可以部分指定設備過濾器。對於沒有指定設備過濾器的遠程任務，所有設備都對其可見。