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

Remove the condition from reconstruction loss

Browse Source
This commit is contained in:
henryruhs
2025-02-23 19:10:38 +01:00
parent 94480e16eb
commit 5bba2a1c69
1 changed files with 2 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
face_swapper/src/models/loss.py
+2 -6
View File
@@ -79,12 +79,8 @@ class ReconstructionLoss(nn.Module):
temp_tensors = []
for _source_tensor, _target_tensor in zip(source_tensor, target_tensor):
temp_tensor = self.mse_loss(_source_tensor, _target_tensor)
if torch.equal(_source_tensor, _target_tensor):
temp_tensors.append(temp_tensor)
else:
temp_tensors.append(temp_tensor * 0)
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
data_range = float(torch.max(output_tensor) - torch.min(output_tensor))