class_docstr: A callback which logs module parameters as histograms to TensorBoard.
  https://pytorch.org/docs/stable/tensorboard.html#torch.utils.tensorboard.writer.SummaryWriter
  
  Args:
      logger: Either a :class:`torchtnt.loggers.tensorboard.TensorBoardLogger`
          or a :class:`torch.utils.tensorboard.SummaryWriter` instance.
code: def on_train_epoch_end(self, state: State, unit: TTrainUnit) -> None:
      writer = self._writer
      if not writer:
          return
  
      step = unit.train_progress.num_steps_completed
      modules = unit.tracked_modules()
      _write_histogram_parameters(writer, modules, step)
  
file_path: tnt/torchtnt/framework/callbacks/tensorboard_parameter_monitor.py
imports: from typing import Dict, Optional, Union
  import torch
  from torch.utils.tensorboard import SummaryWriter
  from torchtnt.framework.callback import Callback
  from torchtnt.framework.state import State
  from torchtnt.framework.unit import TTrainUnit
  from torchtnt.utils.loggers.tensorboard import TensorBoardLogger
label_desc: Use image node_id 2 for calling the TensorBoardParameterMonitor obj's underlying member method code with example usage: obj.on_train_epoch_end(state, unit) without return types