Spec-Zone.ru › TensorFlow 2.9

tf.VariableAggregation

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

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

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

  • NONE: Это значение по умолчанию, вызывающее ошибку при использовании операции обновления переменной с несколькими репликами.
  • SUM: Добавить обновления по всем репликам.
  • MEAN: Взять среднее арифметическое ("среднее значение") обновлений по всем репликам.
  • ONLY_FIRST_REPLICA: Используется, когда каждая реплика выполняет одно и то же обновление, но нам нужно выполнить это обновление только один раз. Например, используется для счётчика глобального шага.
Переменные класса
MEAN <VariableAggregationV2.MEAN: 2>
NONE <VariableAggregationV2.NONE: 0>
ONLY_FIRST_REPLICA <VariableAggregationV2.ONLY_FIRST_REPLICA: 3>
SUM <VariableAggregationV2.SUM: 1>

© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.9/api_docs/python/tf/VariableAggregation

Spec-Zone.ru

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