From b8a0b3d5ea0fc0d173140301fbb240253b705527 Mon Sep 17 00:00:00 2001 From: Franck Mamalet Date: Mon, 15 Apr 2024 07:59:13 +0200 Subject: [PATCH] remove parametrization for lipschitz param when coef_lip==1.0 --- deel/torchlip/modules/module.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/deel/torchlip/modules/module.py b/deel/torchlip/modules/module.py index eaf0a73..ea35fad 100644 --- a/deel/torchlip/modules/module.py +++ b/deel/torchlip/modules/module.py @@ -93,6 +93,8 @@ def __init__(self, coefficient_lip: float = 1.0): def apply_lipschitz_factor(self): """Multiply the layer weights by a lipschitz factor.""" + if self._coefficient_lip == 1.0: + return parametrize.register_parametrization( self, "weight", _LipschitzCoefMultiplication(self._coefficient_lip) )