View source on GitHub
|
Returns the current tf.distribute.Strategy object.
tf.distribute.get_strategy() -> 'StrategyBase'
Typically only used in a cross-replica context:
if tf.distribute.in_cross_replica_context():
strategy = tf.distribute.get_strategy()
...
Returns | |
|---|---|
A tf.distribute.Strategy object. Inside a with strategy.scope() block,
it returns strategy, otherwise it returns the default (single-replica)
tf.distribute.Strategy object.
|