From 746cc86d525d12c21d59c0936f901ea79233aff2 Mon Sep 17 00:00:00 2001
From: henryruhs <info@henryruhs.com>
Date: Wed, 12 Feb 2025 12:51:45 +0100
Subject: [PATCH] Fix monitor

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

diff --git a/embedding_converter/src/training.py b/embedding_converter/src/training.py
index 8e8b086..9e02685 100644
--- a/embedding_converter/src/training.py
+++ b/embedding_converter/src/training.py
@@ -94,7 +94,7 @@ def create_trainer() -> Trainer:
 		callbacks =
 		[
 			ModelCheckpoint(
-				monitor = 'train_loss',
+				monitor = 'loss_training',
 				dirpath = output_directory_path,
 				filename = output_file_pattern,
 				every_n_epochs = 10,