From e8cc2bfff167d603a9dd67fbd5ddd3a5fbca719e Mon Sep 17 00:00:00 2001 From: harisreedhar Date: Tue, 25 Feb 2025 13:59:02 +0530 Subject: [PATCH] add gaze loss --- 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 5e0f844..56cacb4 100644 --- a/face_swapper/src/training.py +++ b/face_swapper/src/training.py @@ -67,7 +67,7 @@ class FaceSwapperTrainer(lightning.LightningModule): identity_loss, weighted_identity_loss = self.identity_loss(generator_output_tensor, source_tensor) pose_loss, weighted_pose_loss = self.pose_loss(target_tensor, generator_output_tensor) gaze_loss, weighted_gaze_loss = self.gaze_loss(target_tensor, generator_output_tensor) - generator_loss = weighted_adversarial_loss + weighted_attribute_loss + weighted_reconstruction_loss + weighted_identity_loss + weighted_pose_loss + generator_loss = weighted_adversarial_loss + weighted_attribute_loss + weighted_reconstruction_loss + weighted_identity_loss + weighted_pose_loss + weighted_gaze_loss generator_optimizer.zero_grad() self.manual_backward(generator_loss)