benchmarks/model_benchmark/benchmark_utils.py/BenchmarkMetricsCallback/__init__
class BenchmarkMetricsCallback:    def __init__(self, start_batch=1, stop_batch=None):
        self.start_batch = start_batch
        self.stop_batch = stop_batch

        # Store the throughput of each epoch.
        self.state = {"throughput": []}