remove parametrization for lipschitz param when coef_lip==1.0

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