Move validation on top (important)

Author: qubvel

segmentation_models_pytorch/decoders/unet/model.py

@@ -121,6 +121,15 @@ def __init__(
     ):
         super().__init__()
 
+        decoder_use_batchnorm = kwargs.pop("decoder_use_batchnorm", None)
+        if decoder_use_batchnorm is not None:
+            warnings.warn(
+                "The usage of decoder_use_batchnorm is deprecated. Please modify your code for decoder_use_norm",
+                DeprecationWarning,
+                stacklevel=2,
+            )
+            decoder_use_norm = decoder_use_batchnorm
+
         self.encoder = get_encoder(
             encoder_name,
             in_channels=in_channels,
@@ -131,15 +140,6 @@ def __init__(
 
         add_center_block = encoder_name.startswith("vgg")
 
-        decoder_use_batchnorm = kwargs.pop("decoder_use_batchnorm", None)
-        if decoder_use_batchnorm is not None:
-            warnings.warn(
-                "The usage of decoder_use_batchnorm is deprecated. Please modify your code for use_norm",
-                DeprecationWarning,
-                stacklevel=2,
-            )
-            decoder_use_norm = decoder_use_batchnorm
-
         self.decoder = UnetDecoder(
             encoder_channels=self.encoder.out_channels,
             decoder_channels=decoder_channels,