Spec-Zone.ru › TensorFlow 2.4

tf.VariableAggregation

Просмотреть исходный код на GitHub

Указывает, как будет агрегирована распределённая переменная.

tf.distribute.Strategy распределяет модель, создавая несколько копий (называемых "репликами"), действующих параллельно на разных элементах входной пакетной обработки. При выполнении некоторой операции обновления переменных, например, var.assign_add(x), в модели нам необходимо определить, как комбинировать различные значения для x, вычисленные в разных репликах.

  • NONE: Это значение по умолчанию, которое выведет ошибку, если вы используете операцию обновления переменной с несколькими репликами.
  • SUM: Добавить обновления по всем репликам.
  • MEAN: Возьмите арифметическое среднее ("среднее значение") обновлений по всем репликам.
  • ONLY_FIRST_REPLICA: Используется, когда каждая реплика выполняет одно и то же обновление, но мы хотим выполнить обновление только один раз. Применяется, например, для счётчика глобальных шагов.
Переменные класса
MEAN tf.VariableAggregation
NONE tf.VariableAggregation
ONLY_FIRST_REPLICA tf.VariableAggregation
SUM tf.VariableAggregation

© 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/r2.4/api_docs/python/tf/VariableAggregation

Spec-Zone.ru

Настройки Оффлайн Что нового Помощь О нас
Spec-Zone .ru
спецификации, руководства, описания, API