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