diff --git a/lib/onnx/onnx.ml b/lib/onnx/onnx.ml index 5dd4c592289f0cb86235ad7ea7d64bdfb14db9d9..6a0ea0a963cc78954c74c3141c492c5bee87bf6c 100644 --- a/lib/onnx/onnx.ml +++ b/lib/onnx/onnx.ml @@ -119,7 +119,7 @@ let produce_cfg (g : Oproto.Onnx.GraphProto.t) = | "MaxPool" -> NCFG.Node.MaxPool | "Conv" -> NCFG.Node.Conv | "Identity" -> NCFG.Node.Identity - | _ -> raise (ParseError ("Unsupported ONNX operator: " ^ o))) + | _ -> raise (ParseError ("Unsupported ONNX operator " ^ o))) in List.fold ~f:(fun acc n -> get_node_operator_cfg n :: acc) ~init:[] ns in diff --git a/src/language.ml b/src/language.ml index 722b15a59f91b9170a375fe0b9ed989276e68492..35cde2dbe6e0eb3fc4b2cb4d19bee8e57680333a 100644 --- a/src/language.ml +++ b/src/language.ml @@ -95,7 +95,14 @@ let onnx_parser env _ filename _ = match model with | Error s -> Loc.errorm "%s" s | Ok { n_inputs; n_outputs; nier } -> - let nier = Result.ok nier (* TODO: Warn about parsing errors? *) in + let nier = + match nier with + | Error msg -> + Logs.warn (fun m -> + m "Cannot build network intermediate representation:@ %s" msg); + None + | Ok nier -> Some nier + in register_nn_as_tuple n_inputs n_outputs filename nier env let ovo_parser env _ filename _ =