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

Different naming

Browse Source
This commit is contained in:
henryruhs
2025-02-26 00:03:57 +01:00
parent 8c24c9ec27
commit 7ce9d27097
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
face_swapper/src/models/loss.py
+2 -2
View File
@@ -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