From dbe931e9504a1c2910c77b41235a953c923272a0 Mon Sep 17 00:00:00 2001 From: henryruhs Date: Fri, 7 Mar 2025 14:04:24 +0100 Subject: [PATCH] Fix naming --- face_swapper/src/models/loss.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/face_swapper/src/models/loss.py b/face_swapper/src/models/loss.py index 1248e1b..6d7a0e5 100644 --- a/face_swapper/src/models/loss.py +++ b/face_swapper/src/models/loss.py @@ -107,7 +107,7 @@ class MotionLoss(nn.Module): def __init__(self, config_parser : ConfigParser, motion_extractor : MotionExtractorModule): super().__init__() self.config_pose_weight = config_parser.getfloat('training.losses', 'pose_weight') - self.expression_weight = config_parser.getfloat('training.losses', 'expression_weight') + self.config_expression_weight = config_parser.getfloat('training.losses', 'expression_weight') self.motion_extractor = motion_extractor self.mse_loss = nn.MSELoss()