Fix monitor

This commit is contained in:
henryruhs
2025-02-12 12:52:38 +01:00
parent 746cc86d52
commit 11c038cb81
+1 -1
View File
@@ -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,