code: def on_train_epoch_end(self, model):
      for metric in model.metrics["train"]:
          self.writer.add_scalar(
              f"train/{metric}",
              model.metrics["train"][metric],
              model.current_epoch,
          )
  
imports: from torch.utils.tensorboard import SummaryWriter
  from tez.callbacks import Callback
label_desc: Use image node_id 2 for calling the TensorBoardLogger obj's underlying member method code with example usage: obj.on_train_epoch_end(model) without return types