diff --git a/extra/bench_log.py b/extra/bench_log.py index 72a23fe13e..de5ddcb685 100644 --- a/extra/bench_log.py +++ b/extra/bench_log.py @@ -15,7 +15,7 @@ class BenchEvent(Enum): STEP = "step" FULL = "full" MLPERF_INIT = "mlperf_init" - MLPERF_RUN = "mlperf_setup" + MLPERF_RUN = "mlperf_run" class InstantBenchEvent(Enum): GFLOPS = "gflops"