tf.VariableSynchronization
Указывает, когда будет синхронизирована распределенная переменная.
-
AUTO: Синхронизация определяется текущим значениемDistributionStrategy(например, сMirroredStrategyэто будетON_WRITE). -
NONE: Указывает, что будет существовать только одна копия переменной, поэтому синхронизация не требуется. -
ON_WRITE: Указывает, что переменная будет обновляться на всех устройствах при каждой записи. -
ON_READ: Указывает, что при чтении переменная будет агрегироваться на всех устройствах (например, при создании контрольной точки или вычислении операции, использующей эту переменную).Пример:
>>> temp_grad=[tf.Variable([0.], trainable=False, ... synchronization=tf.VariableSynchronization.ON_READ, ... aggregation=tf.VariableAggregation.MEAN ... )]
| Переменные класса | |
|---|---|
| AUTO | <VariableSynchronization.AUTO: 0> |
| NONE | <VariableSynchronization.NONE: 1> |
| ON_READ | <VariableSynchronization.ON_READ: 3> |
| ON_WRITE | <VariableSynchronization.ON_WRITE: 2> |
© 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/api_docs/python/tf/VariableSynchronization