Skip to content

GitLab

Switch branch/tag
Find file BlameHistoryPermalink
  • Ayush Dubey's avatar
    Add multi-worker benchmarks to Keras ResNet model. · ff6c3b1e
    Ayush Dubey authored 
    Also add `worker_hosts` and `task_index` flags.  These flags enable running the
model over multiple hosts by passing the cluster information via command line.

Setting `TF_CONFIG` will continue to work.

PiperOrigin-RevId: 263825245
    ff6c3b1e
_distribution.py 1.82 KB