From ff630bdf1809f3220c13fb69a0dcd0195f6922dc Mon Sep 17 00:00:00 2001 From: Adam Hines Date: Sat, 13 Jul 2024 10:40:04 +1000 Subject: [PATCH] Update loggers.py --- vprtempo/src/loggers.py | 16 +++++++++++++--- 1 file changed, 13 insertions(+), 3 deletions(-) diff --git a/vprtempo/src/loggers.py b/vprtempo/src/loggers.py index 8f7e5cb..1c75914 100644 --- 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 \ No newline at end of file + return logger, output_folder