diffusion_models.utils.tensorboard
==================================

.. py:module:: diffusion_models.utils.tensorboard




Module Contents
---------------

.. py:class:: TensorboardManager(log_name = None)

   A tensorboard manager for simplified tensorboard logging.

   :param log_name: The name of the tensorboard log run.


   .. py:attribute:: log_directory

      The directory where tensorboard logs are saved.


   .. py:attribute:: summary_writer

      The tensorboard summary writer.


   .. py:method:: log_metrics(metrics, global_step)

      Log metrics to tensorboard.

      :param metrics: A dictionary mapping metric names to values.
      :param global_step: The step at which the metrics are recorded.



   .. py:method:: log_images(tag, images, timestep)

      Log images to tensorboard.

      :param tag: The name to give the images in tensorboard.
      :param images: A tensor representing the images to log.
      :param timestep: The timestep at which the images are produced.