From fffdf02b932e9147c790aa73a0acfbf31ba12616 Mon Sep 17 00:00:00 2001 From: shahules786 Date: Fri, 30 Sep 2022 15:17:58 +0530 Subject: [PATCH] valid monitor fix --- cli/train.py | 11 ++++++----- 1 file changed, 6 insertions(+), 5 deletions(-) diff --git a/cli/train.py b/cli/train.py index 88e513a..a5c83f0 100644 --- a/cli/train.py +++ b/cli/train.py @@ -4,7 +4,7 @@ from hydra.utils import instantiate from omegaconf import DictConfig from pytorch_lightning.callbacks import ModelCheckpoint, EarlyStopping from pytorch_lightning.loggers import MLFlowLogger - +os.environ["HYDRA_FULL_ERROR"] = "1" @hydra.main(config_path="train_config",config_name="config") def main(config: DictConfig): @@ -20,14 +20,15 @@ def main(config: DictConfig): model = instantiate(config.model,dataset=dataset,lr=parameters.get("lr"), loss=parameters.get("loss"), metric = parameters.get("metric")) + direction = model.valid_monitor checkpoint = ModelCheckpoint( - dirpath="",filename="model",monitor="valid_loss",verbose=False, - mode="min",every_n_epochs=1 + dirpath="",filename="model",monitor="val_loss",verbose=False, + mode=direction,every_n_epochs=1 ) callbacks.append(checkpoint) early_stopping = EarlyStopping( - monitor="valid_loss", - mode="min", + monitor="val_loss", + mode=direction, min_delta=0.0, patience=100, strict=True,