From 8f0ee4935b158024560d6cbe569716112ca42898 Mon Sep 17 00:00:00 2001
From: henryruhs <info@henryruhs.com>
Date: Wed, 12 Mar 2025 22:07:39 +0100
Subject: [PATCH] Remove flag

---
 face_swapper/src/training.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/face_swapper/src/training.py b/face_swapper/src/training.py
index 14bb67e..5ebde71 100644
--- a/face_swapper/src/training.py
+++ b/face_swapper/src/training.py
@@ -123,7 +123,7 @@ class FaceSwapperTrainer(LightningModule):
 		mask_loss = self.mask_loss(target_tensor, mask_tensor)
 
 		self.toggle_optimizer(generator_optimizer)
-		self.manual_backward(generator_loss, retain_graph = True)
+		self.manual_backward(generator_loss)
 		if do_update:
 			generator_optimizer.step()
 			generator_optimizer.zero_grad()