From 11c038cb8101c314f084022531797989d6bd8ebb Mon Sep 17 00:00:00 2001
From: henryruhs <info@henryruhs.com>
Date: Wed, 12 Feb 2025 12:52:38 +0100
Subject: [PATCH] Fix monitor

---
 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 fc20ead..381cfff 100644
--- a/face_swapper/src/training.py
+++ b/face_swapper/src/training.py
@@ -92,7 +92,7 @@ def create_trainer() -> Trainer:
 		callbacks =
 		[
 			ModelCheckpoint(
-				monitor = 'l_G',
+				monitor = 'loss_generator',
 				dirpath = output_directory_path,
 				filename = output_file_pattern,
 				every_n_train_steps = 1000,