Track and Visualize Experiments
Track and Visualize Experiments
More details are here: Track and Visualize Experiments
Track audio and other artifacts
To track other artifacts, such as histograms or model topology graphs first select one of the many loggers supported by Lightning
from lightning.pytorch import loggers as pl_loggers
tensorboard = pl_loggers.TensorBoardLogger(save_dir="")
trainer = Trainer(logger=tensorboard)
then access the logger’s API directly
def training_step(self):
tensorboard = self.logger.experiment
tensorboard.add_image()
tensorboard.add_histogram(...)
tensorboard.add_figure(...)
More details are here: Track and Visualize Experiments
Libs: