Distribution Strategies Example BaseCollectiveExecutor::StartAbort Out of range

[runzhliu]

Hi, I'm testing Distribution Strategies Example on my own K8S cluster, there is an unexpected error before the end of the task.

However, when I run keras_model_to_estimator.py locally, it ends normally.