From f058456a4043a9356a0847095203218e694a9681 Mon Sep 17 00:00:00 2001 From: Lenz Fiedler Date: Mon, 7 Oct 2024 18:15:28 +0200 Subject: [PATCH] Automatically added during training metric --- mala/common/parameters.py | 12 ++---------- 1 file changed, 2 insertions(+), 10 deletions(-) diff --git a/mala/common/parameters.py b/mala/common/parameters.py index c9b1b826c..d15576f1a 100644 --- a/mala/common/parameters.py +++ b/mala/common/parameters.py @@ -780,6 +780,8 @@ def during_training_metric(self, value): "Currently, MALA can only operate with the " '"ldos" metric for ddp runs.' ) + if value not in self.validation_metrics: + self.validation_metrics.append(value) self._during_training_metric = value @property @@ -807,16 +809,6 @@ def after_training_metric(self, value): ) self._after_training_metric = value - @during_training_metric.setter - def during_training_metric(self, value): - if value != "ldos": - if self._configuration["ddp"]: - raise Exception( - "Currently, MALA can only operate with the " - '"ldos" metric for ddp runs.' - ) - self._during_training_metric = value - @property def use_graphs(self): """