diff --git a/optimum_benchmark/scenarios/energy_star/scenario.py b/optimum_benchmark/scenarios/energy_star/scenario.py index 37fc87ec..706feaa3 100644 --- a/optimum_benchmark/scenarios/energy_star/scenario.py +++ b/optimum_benchmark/scenarios/energy_star/scenario.py @@ -85,10 +85,12 @@ def run(self, backend: Backend[BackendConfigT]) -> BenchmarkReport: ) self.init_trackers(backend) - self.run_model_loading_tracking(backend) + self.run_dataset_loading_tracking(backend) self.run_dataset_preprocessing_tracking(backend) + self.run_model_loading_tracking(backend) + self.logger.info("\t+ Preparing sample inputs for model warmup") self.sample_inputs = self.dataset[: self.config.input_shapes["batch_size"]] self.sample_inputs = backend.prepare_inputs(self.sample_inputs)