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