From af09ee7ff3c0c020e32082772674e9b99dbcac65 Mon Sep 17 00:00:00 2001 From: henryruhs Date: Tue, 11 Mar 2025 14:39:47 +0100 Subject: [PATCH] Fix config --- face_swapper/src/training.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/face_swapper/src/training.py b/face_swapper/src/training.py index 0f8ff94..3d25597 100644 --- a/face_swapper/src/training.py +++ b/face_swapper/src/training.py @@ -32,7 +32,7 @@ class FaceSwapperTrainer(LightningModule): self.config_embedder_path = config_parser.get('training.model', 'embedder_path') self.config_gazer_path = config_parser.get('training.model', 'gazer_path') self.config_motion_extractor_path = config_parser.get('training.model', 'motion_extractor_path') - self.config_face_parser_path = config_parser.get('training.model', 'parser_path') + self.config_face_parser_path = config_parser.get('training.model', 'face_parser_path') self.config_accumulate_size = config_parser.getfloat('training.trainer', 'accumulate_size') self.config_learning_rate = config_parser.getfloat('training.trainer', 'learning_rate') self.config_preview_frequency = config_parser.getint('training.trainer', 'preview_frequency')