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

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




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

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

   .. py:attribute:: log_directory


   .. py:attribute:: summary_writer


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


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