tf.contrib.training.RandomStrategy
Возвращает случайную задачу PS для размещения операции.
tf.contrib.training.RandomStrategy(
num_ps_tasks, seed=0
)
Это может работать лучше, чем по умолчанию круговое размещение, если у вас большое количество переменных. В зависимости от вашей архитектуры и количества серверов параметров, круговое размещение может привести к ситуациям, когда все переменные одного типа размещаются на одной задаче PS, что может привести к проблемам с конкуренцией.
Эта стратегия использует хеш-функцию для имени каждой операции для детерминированного размещения.
Методы
__call__
__call__(
op
)
Выбирает индекс задачи ps для заданной Operation.
© 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/contrib/training/RandomStrategy