From 35ce13c7aa49981486135b8a174219be4e260649 Mon Sep 17 00:00:00 2001
From: Leonard Hackel <l.hackel@tu-berlin.de>
Date: Mon, 22 May 2023 15:03:54 +0200
Subject: [PATCH] changing monitor to use correct metric in checkpointing
---
train_lit4rsvqa.py | 6 ++----
1 file changed, 2 insertions(+), 4 deletions(-)
diff --git a/train_lit4rsvqa.py b/train_lit4rsvqa.py
index 6117e57..4189a69 100644
--- a/train_lit4rsvqa.py
+++ b/train_lit4rsvqa.py
@@ -290,7 +290,7 @@ def main(
monitor_str = "AA"
# checkpointing
checkpoint_callback = ModelCheckpoint(
- monitor="val/f1",
+ monitor=monitor,
dirpath="./checkpoints",
filename=f"{wandb_logger.experiment.name}-seed=" +
str(seed) + "-epoch={epoch:03d}-" + f"{monitor_str}" + "={" +
@@ -300,8 +300,6 @@ def main(
mode="max",
save_last=True
)
- early_stopping_callback = EarlyStopping(monitor=monitor, min_delta=0.00,
- patience=25, verbose=False, mode="max")
lr_monitor = LearningRateMonitor(logging_interval='step')
trainer = pl.Trainer(
@@ -310,7 +308,7 @@ def main(
log_every_n_steps=5,
logger=wandb_logger,
check_val_every_n_epoch=2,
- callbacks=[checkpoint_callback, early_stopping_callback, lr_monitor],
+ callbacks=[checkpoint_callback, lr_monitor],
)
--
GitLab