import tensorflow as tf



# TRAINING PROGRESS EVENTS


  def _on_training_start(self, sess):
    # Writers and savers
    self.summary_writer = tf.summary.FileWriter(FLAGS.logdir, sess.graph)
    self.saver = tf.train.Saver()
    self._build_embedding_saver(sess)
    self._restore_model(sess)
    # Loss summaries
    self._build_summaries()

    self.epoch_stats = get_stats_template()
    self.stats = Bunch(
