CalvinBackup/facefusion-labs
1
0
Fork 0
mirror of https://github.com/facefusion/facefusion-labs.git synced 2026-04-19 15:56:37 +02:00
Code Issues Packages Projects Releases Wiki Activity

Rename temp1 with positive and temp2 with negative

Browse Source
This commit is contained in:
henryruhs
2025-02-23 09:35:18 +01:00
parent e75a3c58f9
commit fad38da864
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
face_swapper/src/models/loss.py
+3 -3
View File
@@ -28,9 +28,9 @@ class DiscriminatorLoss(nn.Module):
negative_tensor = torch.relu(1 - discriminator_source_tensor[0]).mean(dim = [ 1, 2, 3 ])
negative_tensors.append(negative_tensor)
discriminator_positive_loss = torch.stack(positive_tensors).mean()
discriminator_negative_loss = torch.stack(negative_tensors).mean()
discriminator_loss = (discriminator_positive_loss + discriminator_negative_loss) * 0.5
positive_loss = torch.stack(positive_tensors).mean()
negative_loss = torch.stack(negative_tensors).mean()
discriminator_loss = (positive_loss + negative_loss) * 0.5
return discriminator_loss