Update loggers.py

QVPR · Jul 13, 2024 · ff630bd · ff630bd
1 parent b1bc693
commit ff630bd
Showing 1 changed file with 13 additions and 3 deletions.
diff --git a/vprtempo/src/loggers.py b/vprtempo/src/loggers.py
@@ -9,8 +9,13 @@ def model_logger():
     Configure the model logger
     """   
     now = datetime.now()
-    output_folder = './vprtempo/output/' + now.strftime("%d%m%y-%H-%M-%S")
+    output_base_folder = './vprtempo/output/'
+    output_folder = output_base_folder + now.strftime("%d%m%y-%H-%M-%S")
+
+    # Create the base output folder if it does not exist
+    os.makedirs(output_base_folder, exist_ok=True)
 
+    # Create the specific output folder
     os.mkdir(output_folder)
     # Create the logger
     logger = logging.getLogger("VPRTempo")
@@ -56,8 +61,13 @@ def model_logger_quant():
     """   
 
     now = datetime.now()
-    output_folder = './vprtempo/output/' + now.strftime("%d%m%y-%H-%M-%S")
+    output_base_folder = './vprtempo/output/'
+    output_folder = output_base_folder + now.strftime("%d%m%y-%H-%M-%S")
+
+    # Create the base output folder if it does not exist
+    os.makedirs(output_base_folder, exist_ok=True)
 
+    # Create the specific output folder
     os.mkdir(output_folder)
     # Create the logger
     logger = logging.getLogger("VPRTempo")
@@ -91,4 +101,4 @@ def model_logger_quant():
     logger.info('Current device is: CPU')
     logger.info('')
 
-    return logger, output_folder
+    return logger, output_folder