suppress logging while calling MLPerfLog(), which is too chatty

krai · Jan 16, 2024 · 5f4cb03 · 5f4cb03
1 parent d18edc8
commit 5f4cb03
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/base_power_measurement_experiment/code_axs.py b/base_power_measurement_experiment/code_axs.py
@@ -2,6 +2,7 @@
 
 import datetime
 import sys
+import logging
 
 
 def load_lines(fname):
@@ -17,8 +18,14 @@ def avg_power( mlperf_log_parser_path, server_timezone_sec, client_timezone_sec,
     server_timezone = datetime.timedelta(seconds=server_timezone_sec)
     client_timezone = datetime.timedelta(seconds=client_timezone_sec)
 
+    logger = logging.getLogger()
+    orig_level = logger.level
+    logger.setLevel(logging.CRITICAL)
+
     mlperf_log = MLPerfLog(detail_log_path)
 
+    logger.setLevel(orig_level)
+
     datetime_format = '%m-%d-%Y %H:%M:%S.%f'
     power_begin = datetime.datetime.strptime(mlperf_log["power_begin"], datetime_format) + client_timezone
     power_end = datetime.datetime.strptime(mlperf_log["power_end"], datetime_format) + client_timezone