Python 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 範圍之外調用，以避免不正確的嵌套。