code: def on_epoch_end(self, epoch, logs=None):
      epoch_run_time = time.time() - self.epoch_start
      if hvd.rank() == 0:
          print("epoch: %d time_taken: %.1f" % (epoch, epoch_run_time))
  
file_path: DALI/docs/examples/use_cases/tensorflow/resnet-n/nvutils/runner.py
imports: from nvutils import image_processing
  from nvutils import common
  from distutils.version import StrictVersion
  import tensorflow
  import keras
  import os
  import time
  import re
  import horovod.tensorflow.keras
  from keras import backend
label_desc: Use image node_id 5 for calling the _ProfileKerasFitCallback obj's underlying member method code with example usage: obj.on_epoch_end(epoch, logs) without return types