diff --git a/optimum_benchmark/launchers/process/launcher.py b/optimum_benchmark/launchers/process/launcher.py index c08061a5..538e84f3 100644 --- a/optimum_benchmark/launchers/process/launcher.py +++ b/optimum_benchmark/launchers/process/launcher.py @@ -42,9 +42,6 @@ def launch(self, worker: Callable, *worker_args) -> BenchmarkReport: while not process_context.join(): pass - # restore the original logging configuration - setup_logging(log_level) - report: BenchmarkReport = queue.get() return report diff --git a/optimum_benchmark/launchers/torchrun/launcher.py b/optimum_benchmark/launchers/torchrun/launcher.py index d5351a34..c86630cb 100644 --- a/optimum_benchmark/launchers/torchrun/launcher.py +++ b/optimum_benchmark/launchers/torchrun/launcher.py @@ -57,9 +57,6 @@ def launch(self, worker: Callable, *worker_args) -> Dict[str, Any]: entrypoint=entrypoint, args=(worker, queue, lock, log_level, *worker_args), config=launch_config ) - # restore the original logging configuration - setup_logging(log_level) - reports: List[BenchmarkReport] = [] while not queue.empty(): reports.append(queue.get())