do model loading after dataset loading and processing for better memo…

…ry measurments.
huggingface · Dec 4, 2024 · d36a9f2 · d36a9f2
1 parent e051a83
commit d36a9f2
Showing 1 changed file with 3 additions and 1 deletion.
diff --git 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)