CalvinBackup/facefusion-labs
1
0 Fork 0
mirror of https://github.com/facefusion/facefusion-labs.git synced 2026-10-03 08:16:49 +02:00
Code Issues Packages Projects Releases Wiki Activity

add gaze loss

Browse Source
This commit is contained in:
harisreedhar authored and henryruhs committed 2025-03-11 14:43:10 +01:00
1 parent a951d700fc
commit e8cc2bfff1
1 file changed
+1 -1
Download Patch File Download Diff File Expand all files Collapse all files
face_swapper/src/training.py
+1 -1
View File
@@ -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)