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

changes

Browse Source
This commit is contained in:
harisreedhar
2025-04-15 14:10:58 +05:30
parent 4a319ec9bd
commit 128726701b
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
@@ -64,8 +64,8 @@ class CycleLoss(nn.Module):
temp_tensors.append(temp_tensor)
feature_loss = torch.stack(temp_tensors).mean()
mae_loss = self.mae_loss(target_tensor, cycle_tensor)
cycle_loss = (feature_loss + mae_loss) * 0.5
reconstruction_loss = self.mae_loss(target_tensor, cycle_tensor)
cycle_loss = (feature_loss + reconstruction_loss) * 0.5
weighted_feature_loss = cycle_loss * self.config_cycle_weight
return cycle_loss, weighted_feature_loss