Skip to content
Snippets Groups Projects
Commit 3911f85d authored by Michele Alberti's avatar Michele Alberti
Browse files

Add some info logs about model format checking.

parent f74e2942
No related branches found
No related tags found
No related merge requests found
...@@ -180,6 +180,7 @@ let retrieve_nnet_metadata filename = ...@@ -180,6 +180,7 @@ let retrieve_nnet_metadata filename =
let build ~filename = let build ~filename =
let open Result in let open Result in
Logs.info (fun m -> m "Checking format of model file `%s'." filename);
if Sys.file_exists filename if Sys.file_exists filename
then then
begin begin
...@@ -189,6 +190,9 @@ let build ~filename = ...@@ -189,6 +190,9 @@ let build ~filename =
retrieve_nnet_metadata filename >>= fun nnet -> retrieve_nnet_metadata filename >>= fun nnet ->
Ok (Nnet nnet) Ok (Nnet nnet)
end >>= fun format -> end >>= fun format ->
Logs.info (fun m ->
m "Model format recognized as `%s'."
(match format with Onnx -> "ONNX" | Nnet _ -> "NNet"));
Ok { format; filename } Ok { format; filename }
else else
Error (Format.sprintf "No such file `%s'." filename) Error (Format.sprintf "No such file `%s'." filename)
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment