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