From 3b7d3b6688d49a3324c8ec2926afd78e8a9be17f Mon Sep 17 00:00:00 2001
From: henryruhs <info@henryruhs.com>
Date: Sat, 22 Feb 2025 14:30:52 +0100
Subject: [PATCH] Introduce new ReconstructionLoss class

---
 face_swapper/src/models/loss.py | 1 -
 1 file changed, 1 deletion(-)

diff --git a/face_swapper/src/models/loss.py b/face_swapper/src/models/loss.py
index 1f85614..356f80b 100644
--- a/face_swapper/src/models/loss.py
+++ b/face_swapper/src/models/loss.py
@@ -147,7 +147,6 @@ class ReconstructionLoss(torch.nn.Module):
 
 	def calc(self, source_tensor : Tensor, target_tensor : Tensor, output_tensor : Tensor) -> Tensor:
 		batch_size = CONFIG.getint('training.loader', 'batch_size')
-
 		loss_tensor = torch.pow(output_tensor - target_tensor, 2).reshape(batch_size, -1)
 		loss_tensor = torch.mean(loss_tensor, dim = 1) * 0.5