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 范围之外调用，以避免不正确的嵌套。