Skip to content
Snippets Groups Projects

Add distributed training support with dask

Merged Amir MOHAMMADI requested to merge inception-resenet-change into master
Files
15
import tensorflow as tf
def strategy_fn():
print("Creating MirroredStrategy strategy.")
strategy = tf.distribute.MirroredStrategy()
print("MirroredStrategy strategy created.")
print("Number of devices: {}".format(strategy.num_replicas_in_sync))
return strategy
Loading