import tensorflow as tf

        if not self.graph.finalized:
            self.graph.finalize()

        tf.logging.info('Start training')
        for i in range(iterations):
            loss, summaries, _ = self.sess.run(
