From 3911f85d53fdf06a236acc3d23923125f5f0e092 Mon Sep 17 00:00:00 2001
From: Michele Alberti <michele.alberti@cea.fr>
Date: Wed, 24 Mar 2021 18:43:32 +0100
Subject: [PATCH] Add some info logs about model format checking.
---
model.ml | 4 ++++
1 file changed, 4 insertions(+)
diff --git a/model.ml b/model.ml
index 8fc7eb5..a9d8bdc 100644
--- a/model.ml
+++ b/model.ml
@@ -180,6 +180,7 @@ let retrieve_nnet_metadata filename =
let build ~filename =
let open Result in
+ Logs.info (fun m -> m "Checking format of model file `%s'." filename);
if Sys.file_exists filename
then
begin
@@ -189,6 +190,9 @@ let build ~filename =
retrieve_nnet_metadata filename >>= fun nnet ->
Ok (Nnet nnet)
end >>= fun format ->
+ Logs.info (fun m ->
+ m "Model format recognized as `%s'."
+ (match format with Onnx -> "ONNX" | Nnet _ -> "NNet"));
Ok { format; filename }
else
Error (Format.sprintf "No such file `%s'." filename)
--
GitLab