View source on GitHub
|
Set a tf.distribute.Strategy as current without 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()
is equivalent to:
with strategy1.scope():
f()
with strategy2.scope():
g()
h()
In general, you should use the with strategy.scope(): API, but this
alternative may be convenient in notebooks where you would have to put
each cell in a with strategy.scope(): block.
Args | |
|---|---|
strategy
|
A tf.distribute.Strategy object or None.
|
Raises | |
|---|---|
RuntimeError
|
If called inside a with strategy.scope():.
|