Method on_train_end

tensorflow/python/keras/callbacks_v1.py:453–457

(self, logs=None)

          i += self.batch_size

  def on_train_end(self, logs=None):
    if self._is_profiling:
      profiler.save(self.log_dir, profiler.stop())
      self._is_profiling = False
    self.writer.close()