diff --git a/mlair/model_modules/fully_connected_networks.py b/mlair/model_modules/fully_connected_networks.py index 948d2b06d7bf801c585a4e7193a68cf75ada9e8a..1f965f3c8b4ad997a829ebb643ce0a219cdee013 100644 --- a/mlair/model_modules/fully_connected_networks.py +++ b/mlair/model_modules/fully_connected_networks.py @@ -120,8 +120,8 @@ class FCN(AbstractModelClass): raise AttributeError(f"Given optimizer {optimizer} is not supported in this model class.") def _set_regularizer(self, regularizer, **kwargs): - if regularizer is None: - return regularizer + if regularizer is None or (isinstance(regularizer, str) and regularizer.lower() == "none"): + return None try: reg_name = regularizer.lower() reg = self._regularizer.get(reg_name)