diffusion_models.utils.tensorboard

Module Contents

class TensorboardManager(log_name=None)[source]

A tensorboard manager for simplified tensorboard logging.

Parameters:

log_name (Optional[str]) – The name of the tensorboard log run.

log_directory[source]

The directory where tensorboard logs are saved.

summary_writer[source]

The tensorboard summary writer.

log_metrics(metrics, global_step)[source]

Log metrics to tensorboard.

Parameters:

  • metrics (Dict[str, float]) – A dictionary mapping metric names to values.

  • global_step (int) – The step at which the metrics are recorded.

log_images(tag, images, timestep)[source]

Log images to tensorboard.

Parameters:

  • tag (str) – The name to give the images in tensorboard.

  • images (Tensor) – A tensor representing the images to log.

  • timestep (int) – The timestep at which the images are produced.