From 7ce9d270978989e85f58854a8d38eb02e0f82794 Mon Sep 17 00:00:00 2001 From: henryruhs Date: Wed, 26 Feb 2025 00:03:57 +0100 Subject: [PATCH] Different naming --- face_swapper/src/models/loss.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/face_swapper/src/models/loss.py b/face_swapper/src/models/loss.py index df3343a..a41d76e 100644 --- a/face_swapper/src/models/loss.py +++ b/face_swapper/src/models/loss.py @@ -78,8 +78,8 @@ class ReconstructionLoss(nn.Module): reconstruction_weight = CONFIG.getfloat('training.losses', 'reconstruction_weight') temp_tensors = [] - for _source_tensor, _target_tensor in zip(source_tensor, target_tensor): - temp_tensor = self.mse_loss(_source_tensor, _target_tensor) * torch.equal(_source_tensor, _target_tensor) + for __source_tensor__, __target_tensor__ in zip(source_tensor, target_tensor): + temp_tensor = self.mse_loss(__source_tensor__, __target_tensor__) * torch.equal(__source_tensor__, __target_tensor__) temp_tensors.append(temp_tensor) reconstruction_loss = torch.stack(temp_tensors).mean() * 0.5