code: def on_train_batch_end(self, batch, logs=None):
      if batch == self.stop_batch:
          self.state["benchmark_end"] = time.time()
          throughput = (self.stop_batch - self.start_batch + 1) / (
              self.state["benchmark_end"]
              - self.state["benchmark_begin"]
          )
          self.state["throughput"] = throughput
  
imports: import time
  import numpy
  import tensorflow
  from absl import flags
  import keras
label_desc: Use image node_id 3 for calling the BenchmarkMetricsCallback obj's underlying member method code with example usage: obj.on_train_batch_end(batch, logs) without return types