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

        self.state = {}