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(