import tensorflow as tf

    '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,
    'Wether we will use the ohkm for hard keypoints.')
tf.app.flags.DEFINE_string(
    'data_format', 'channels_first', # 'channels_first' or 'channels_last'
