official/resnet/resnet_run_loop.py · f5bb2af2c2f00714723cc486726e534291206925 · ModelZoo / ResNet50_tensorflow

Shard input for distribution strategy. (#6349) · 04792078

Ayush Dubey authored Mar 19, 2019

* Shard input for distribution strategy.

* Pass in input_context from real input_fn.

* Pass in input_context from real input_fn.

* Make pipeline id base 1 for better readability.

04792078

resnet_run_loop.py 32.9 KB

Replace resnet_run_loop.py