Spec-Zone.ru › TensorFlow 2.3

tf.VariableAggregation

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

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

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

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

Переменные класса

  • MEAN
  • NONE
  • ONLY_FIRST_REPLICA
  • SUM

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

Spec-Zone.ru

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