TensorBoardParameterMonitor

class torchtnt.framework.callbacks.TensorBoardParameterMonitor(logger: Union[TensorBoardLogger, SummaryWriter])

A callback which logs module parameters as histograms to TensorBoard. https://pytorch.org/docs/stable/tensorboard.html#torch.utils.tensorboard.writer.SummaryWriter

Parameters:logger – Either a torchtnt.loggers.tensorboard.TensorBoardLogger or a torch.utils.tensorboard.SummaryWriter instance.

on_train_epoch_end(state: State, unit: TrainUnit[TTrainData]) → None

Hook called after a train epoch ends.