tf.VariableSynchronization
| View source on GitHub |
Указывает, когда распределенная переменная будет синхронизирована.
-
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/versions/r2.9/api_docs/python/tf/VariableSynchronization