code: def on_train_batch_end(
      self, trainer, pl_module, outputs, batch, batch_idx
  ):
      if not self.disabled:
          self.log_img(pl_module, batch, batch_idx, split="train")
  
file_path: NeMo/nemo/collections/multimodal/models/text_to_image/controlnet/util.py
imports: import os
  import numpy
  import torch
  import torchvision
  from PIL import Image
  from pytorch_lightning import Callback
  from pytorch_lightning.utilities.rank_zero import rank_zero_only
label_desc: Use image node_id 5 for calling the ImageLogger obj's underlying member method code with example usage: obj.on_train_batch_end(trainer, pl_module, outputs, batch, batch_idx) without return types