From 128726701b4bd5c5f0e5f7b6af0983dc6aef7e9b Mon Sep 17 00:00:00 2001 From: harisreedhar Date: Tue, 15 Apr 2025 14:10:58 +0530 Subject: [PATCH] changes --- 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 ccbd96a..1cca867 100644 --- a/face_swapper/src/models/loss.py +++ b/face_swapper/src/models/loss.py @@ -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