mirror of
https://github.com/facefusion/facefusion-labs.git
synced 2026-06-25 07:59:55 +02:00
Fix naming
This commit is contained in:
@@ -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()
|
||||
|
||||
|
||||
Reference in New Issue
Block a user