Fix monitor

This commit is contained in:
henryruhs
2025-02-12 12:51:45 +01:00
parent 58c81cd646
commit 746cc86d52
+1 -1
View File
@@ -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,