diff --git a/src/ptbench/scripts/train.py b/src/ptbench/scripts/train.py index 2ef766421b1ffe34d6bbfe9caee60db5f00d256f..9f770b592e13228ac9f47eedbfe9705c3f133a88 100644 --- a/src/ptbench/scripts/train.py +++ b/src/ptbench/scripts/train.py @@ -259,7 +259,7 @@ def train( if hasattr(model, "set_normalizer"): model.set_normalizer(datamodule.train_dataloader()) else: - logger.info( + logger.warning( f"Model {model.name} has no 'set_normalizer' method. No normalization will be applied." )