From 2baa8e0bfa8acb0c7774ed784118f7ff506de0e6 Mon Sep 17 00:00:00 2001
From: dcarron <daniel.carron@idiap.ch>
Date: Wed, 5 Jul 2023 13:02:09 +0200
Subject: [PATCH] Change 'no set_normalizer' logger info to warning
---
src/ptbench/scripts/train.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/src/ptbench/scripts/train.py b/src/ptbench/scripts/train.py
index 2ef76642..9f770b59 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."
)
--
GitLab