From 236447ec226c08825a4cdf9eabb3940b41b427bd Mon Sep 17 00:00:00 2001
From: dcarron <daniel.carron@idiap.ch>
Date: Fri, 3 May 2024 08:45:48 +0200
Subject: [PATCH] [trainer] Hardcode name of loss to monitor
---
src/mednet/engine/trainer.py | 4 +---
1 file changed, 1 insertion(+), 3 deletions(-)
diff --git a/src/mednet/engine/trainer.py b/src/mednet/engine/trainer.py
index e4a2f5ea..5ea8ccae 100644
--- a/src/mednet/engine/trainer.py
+++ b/src/mednet/engine/trainer.py
@@ -93,8 +93,6 @@ def run(
main_pid=os.getpid(),
)
- monitor_key = "loss/validation"
-
# This checkpointer will operate at the end of every validation epoch
# (which happens at each checkpoint period), it will then save the lowest
# validation loss model observed. It will also save the last trained model
@@ -102,7 +100,7 @@ def run(
dirpath=output_folder,
filename=CHECKPOINT_ALIASES["best"],
save_last=True, # will (re)create the last trained model, at every iteration
- monitor=monitor_key,
+ monitor="loss/validation",
mode="min",
save_on_train_epoch_end=True,
every_n_epochs=validation_period, # frequency at which it checks the "monitor"
--
GitLab