Add: debug msg when loading echecker model in fluerr.

Labbeti · Jul 24, 2023 · 9d97040 · 9d97040
1 parent 2eae3fe
commit 9d97040
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/src/aac_metrics/functional/fluerr.py b/src/aac_metrics/functional/fluerr.py
@@ -414,6 +414,12 @@ def __load_pretrain_echecker(
         pylog.debug(f"Loading echecker model from '{file_path}'.")
 
     model_states = torch.load(file_path)
+
+    if verbose >= 2:
+        pylog.debug(
+            f"Loading echecker model type '{model_states['model_type']}' with '{model_states['num_classes']}' classes."
+        )
+
     echecker = BERTFlatClassifier(
         model_type=model_states["model_type"],
         num_classes=model_states["num_classes"],