From 00dccf07b94343d63c9f524f0671ba9957da1021 Mon Sep 17 00:00:00 2001 From: henryruhs Date: Fri, 21 Feb 2025 17:01:22 +0100 Subject: [PATCH] Fix tensor foobar --- face_swapper/src/models/loss.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/face_swapper/src/models/loss.py b/face_swapper/src/models/loss.py index 73db988..bb0b9be 100644 --- a/face_swapper/src/models/loss.py +++ b/face_swapper/src/models/loss.py @@ -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),