import tensorflow as tf

    'epochs_per_eval', 20,
    'The number of training epochs to run between evaluations.')
tf.app.flags.DEFINE_integer(
    'batch_size', 10,
    'Batch size for training and evaluation.')
tf.app.flags.DEFINE_integer(
    'xt_batch_size', 10,
    'Batch size for training and evaluation.')
tf.app.flags.DEFINE_boolean(
    'use_ohkm', True,
