code: def on_train_epoch_end(
      self, trainer: Trainer, pl_module: LightningModule
  ):
      if self.should_benchmark(trainer):
          epoch_time = time.time() - self.epoch_start_time
          self.train_times.append(epoch_time)
          logging.info(
              f"Training-Epoch-{trainer.current_epoch}-Time: {epoch_time} [sec]"
          )
  
file_path: NeMo/examples/multimodal/x_to_nerf/benchmark_callback.py
imports: import time
  from typing import Optional
  from pytorch_lightning import Callback, LightningModule, Trainer
  from nemo.utils import logging
label_desc: Use image node_id 4 for calling the BenchmarkCallback obj's underlying member method code with example usage: obj.on_train_epoch_end(trainer, pl_module) without return types