import tensorflow as tf

  def build_ae_model(self):
    self.input = tf.placeholder(tf.uint8, self.batch_shape, name='input')
    self.target = tf.placeholder(tf.uint8, self.batch_shape, name='target')
    self.step = tf.Variable(0, trainable=False, name='global_step')
