tf.distribute.experimental_set_strategy

Установить tf.distribute.Strategy в качестве текущего без with strategy.scope().

tf.distribute.experimental_set_strategy(
    strategy
)

tf.distribute.experimental_set_strategy(strategy1)
f()
tf.distribute.experimental_set_strategy(strategy2)
g()
tf.distribute.experimental_set_strategy(None)
h()

эквивалентно:

with strategy1.scope():
  f()
with strategy2.scope():
  g()
h()

В общем случае, вы должны использовать with strategy.scope(): API, но этот альтернативный способ может быть удобен в ноутбуках, где вам необходимо поместить каждый блок кода в with strategy.scope(): блок.

Примечание: Это должно вызываться только вне любого контекста TensorFlow, чтобы избежать неправильного вложенного вызова.