From 20aa5114e10f0aec5f64f106f4b8587466614863 Mon Sep 17 00:00:00 2001 From: henryruhs Date: Wed, 12 Mar 2025 16:24:03 +0100 Subject: [PATCH] Extend config --- embedding_converter/src/training.py | 2 +- face_swapper/src/training.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/embedding_converter/src/training.py b/embedding_converter/src/training.py index bf577ee..8333fc8 100644 --- a/embedding_converter/src/training.py +++ b/embedding_converter/src/training.py @@ -95,7 +95,7 @@ def create_trainer() -> Trainer: config_logger_name = CONFIG_PARSER.get('training.trainer', 'logger_name') config_directory_path = CONFIG_PARSER.get('training.output', 'directory_path') config_file_pattern = CONFIG_PARSER.get('training.output', 'file_pattern') - logger = TensorBoardLogger(config_logger_path, name = config_logger_name) + logger = TensorBoardLogger(config_logger_path, config_logger_name) return Trainer( logger = logger, diff --git a/face_swapper/src/training.py b/face_swapper/src/training.py index 754d6b7..f5a62b0 100644 --- a/face_swapper/src/training.py +++ b/face_swapper/src/training.py @@ -214,7 +214,7 @@ def create_trainer() -> Trainer: config_logger_name = CONFIG_PARSER.get('training.trainer', 'logger_name') config_directory_path = CONFIG_PARSER.get('training.output', 'directory_path') config_file_pattern = CONFIG_PARSER.get('training.output', 'file_pattern') - logger = TensorBoardLogger(config_logger_path, name = config_logger_name) + logger = TensorBoardLogger(config_logger_path, config_logger_name) return Trainer( logger = logger,