Check if set_normalizer method is defined in model

src/ptbench/scripts/train.py

@@ -256,7 +256,12 @@ def train(
     # Sets the model normalizer with the unaugmented-train-subset.
     # this call may be a NOOP, if the model was pre-trained and expects
     # different weights for the normalisation layer.
-    model.set_normalizer(datamodule.train_dataloader())
+    if hasattr(model, "set_normalizer"):
+        model.set_normalizer(datamodule.train_dataloader())
+    else:
+        logger.info(
+            f"Model {model.name} has no 'set_normalizer' method. No normalization will be applied."
+        )
 
     # Rebalances the loss criterion based on the relative proportion of class
     # examples available in the training set.  Also affects the validation loss