Fix config

This commit is contained in:
henryruhs
2025-03-11 14:39:47 +01:00
parent 09432d9214
commit af09ee7ff3
+1 -1
View File
@@ -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')