def _reinitialize_iterator(iterator, distribution_strategy=None):
    if distribution_strategy:
        distributed_training_utils.initialize_iterator(iterator,
                                                       distribution_strategy)
    else:
        training_utils.initialize_iterator(iterator)
Esempio n. 2
0
def _reinitialize_iterator(iterator, distribution_strategy=None):
  if distribution_strategy:
    distributed_training_utils.initialize_iterator(
        iterator, distribution_strategy)
  else:
    training_utils.initialize_iterator(iterator)