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


  return logits, loss


class RevNetTest(tf.test.TestCase):

  def setUp(self):
    super(RevNetTest, self).setUp()
    config = config_.get_hparams_cifar_38()
    config.add_hparam("n_classes", 10)
    config.add_hparam("dataset", "cifar-10")
    # Reconstruction could cause numerical error, use double precision for tests
    config.dtype = tf.float64
    config.fused = False  # Fused batch norm does not support tf.float64
    # Reduce the batch size for tests because the OSS version runs
    # in constrained GPU environment with 1-2GB of memory.
