from tensorflow.contrib.eager.python.examples.revnet import config as config_

                             make_iterator,
                             device_and_format,
                             defun=False,
                             execution_mode=None):
    config = config_.get_hparams_imagenet_56()
    with tfe.execution_mode(execution_mode):
      device, data_format = device_and_format
      for batch_size in self._train_batch_sizes():
