From fb4fe5d40b6c842f8ee6cb3ce350e7ec29f9df5a Mon Sep 17 00:00:00 2001 From: mertalev <101130780+mertalev@users.noreply.github.com> Date: Sat, 8 Jun 2024 21:22:38 -0400 Subject: [PATCH] change log message --- machine-learning/app/models/facial_recognition/recognition.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/machine-learning/app/models/facial_recognition/recognition.py b/machine-learning/app/models/facial_recognition/recognition.py index bc362fb156..5ccdc48e34 100644 --- a/machine-learning/app/models/facial_recognition/recognition.py +++ b/machine-learning/app/models/facial_recognition/recognition.py @@ -32,7 +32,7 @@ class FaceRecognizer(InferenceModel): def _load(self) -> ModelSession: session = self._make_session(self.model_path) if isinstance(session, ort.InferenceSession) and not ort_has_batch_dim(session): - log.debug(f"Adding batch dimension to model {self.model_path}") + log.info(f"Adding batch dimension to recognition model {self.model_name}") ort_add_batch_dim(self.model_path, self.model_path) session = self._make_session(self.model_path) self.model = ArcFaceONNX(