CalvinBackup/facefusion-labs
1
0
Fork 0
mirror of https://github.com/facefusion/facefusion-labs.git synced 2026-05-22 23:59:40 +02:00
Code Issues Packages Projects Releases Wiki Activity

Introduce new AdversarialLoss class

Browse Source
This commit is contained in:
henryruhs
2025-02-22 16:26:40 +01:00
parent f2833a32c3
commit 38211f0340
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
face_swapper/src/training.py
+1 -1
View File
@@ -85,7 +85,7 @@ class FaceSwapperTrainer(lightning.LightningModule, FaceSwapperLoss):
adversarial_loss, weighted_adversarial_loss = self.adversarial_loss.calc(discriminator_output_tensors)
reconstruction_loss, weighted_reconstruction_loss = self.reconstruction_loss.calc(source_tensor, target_tensor, generator_output_tensor)
identity_loss, weighted_identity_loss = self.identity_loss.calc(generator_output_tensor, source_tensor)
generator_loss = weighted_adversarial_loss+ weighted_reconstruction_loss + weighted_identity_loss
generator_loss = weighted_adversarial_loss + weighted_reconstruction_loss + weighted_identity_loss
self.log('generator_loss_new', generator_loss, prog_bar = True)
self.log('adversarial_loss_new', adversarial_loss, prog_bar = True)