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