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): """