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

Fix tensor foobar

Browse Source
This commit is contained in:
henryruhs
2025-02-21 17:01:22 +01:00
parent 5a6e3393e2
commit 00dccf07b9
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
face_swapper/src/models/loss.py
+1 -1
View File
@@ -34,7 +34,7 @@ class FaceSwapperLoss:
weight_pose = CONFIG.getfloat('training.losses', 'weight_pose')
weight_gaze = CONFIG.getfloat('training.losses', 'weight_gaze')
source_tensor, target_tensor = batch
is_same_person = torch.tensor(0) if source_tensor == target_tensor else torch.tensor(1)
is_same_person = torch.tensor(0) if torch.equal(source_tensor, target_tensor) else torch.tensor(1)
generator_loss_set =\
{
'loss_adversarial': self.calc_adversarial_loss(discriminator_outputs),