tf.compat.v1.VariableAggregation
Указывает, как будет агрегироваться распределённая переменная.
tf.distribute.Strategy распределяет модель, создавая несколько копий (называемых «репликами»), действующих параллельно с данными на разных элементах входной партии. При выполнении какой-либо операции обновления переменных, например, var.assign_add(x), в модели необходимо определить, как объединить различные значения для x, вычисленные в разных репликах.
-
NONE: Это значение по умолчанию, вызывающее ошибку при использовании операции обновления переменной с несколькими репликами. -
SUM: Добавить обновления по всем репликам. -
MEAN: Взять арифметическое среднее («среднее значение») обновлений по всем репликам. -
ONLY_FIRST_REPLICA: Используется в случае, когда каждая реплика выполняет одно и то же обновление, но обновление должно быть выполнено только один раз. Используется, например, для счётчика глобального шага. -
ONLY_FIRST_TOWER: Устаревшее псевдоним дляONLY_FIRST_REPLICA.
| Переменные класса | |
|---|---|
| MEAN | tf.compat.v1.VariableAggregation |
| NONE | tf.compat.v1.VariableAggregation |
| ONLY_FIRST_REPLICA | tf.compat.v1.VariableAggregation |
| SUM | tf.compat.v1.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/compat/v1/VariableAggregation