summaryrefslogtreecommitdiffstats
path: root/eval
diff options
context:
space:
mode:
authorHenning Baldersheim <balder@yahoo-inc.com>2023-09-04 13:57:58 +0000
committerHenning Baldersheim <balder@yahoo-inc.com>2023-09-04 13:57:58 +0000
commit5f9ef04076bb9ee0e9866380ee5af1a85b21509b (patch)
treec1b3f165731853311aee600238ad8dcec93a00e5 /eval
parent56025c33579feab8bc93cfc10043f2f91257e8c4 (diff)
Probe with optimize as in production.
Diffstat (limited to 'eval')
-rw-r--r--eval/src/apps/analyze_onnx_model/analyze_onnx_model.cpp8
1 files changed, 3 insertions, 5 deletions
diff --git a/eval/src/apps/analyze_onnx_model/analyze_onnx_model.cpp b/eval/src/apps/analyze_onnx_model/analyze_onnx_model.cpp
index 97fb15dad67..864bc7d095d 100644
--- a/eval/src/apps/analyze_onnx_model/analyze_onnx_model.cpp
+++ b/eval/src/apps/analyze_onnx_model/analyze_onnx_model.cpp
@@ -264,18 +264,16 @@ int probe_types() {
Slime result;
auto &root = result.setObject();
auto &types = root.setObject("outputs");
- Onnx model(params["model"].asString().make_string(), Onnx::Optimize::DISABLE);
+ Onnx model(params["model"].asString().make_string(), Onnx::Optimize::ENABLE);
Onnx::WirePlanner planner;
for (const auto & i : model.inputs()) {
auto spec = params["inputs"][i.name].asString().make_string();
auto input_type = ValueType::from_spec(spec);
if (input_type.is_error()) {
if (!params["inputs"][i.name].valid()) {
- throw MyError{fmt("missing type for model input '%s'",
- i.name.c_str())};
+ throw MyError(fmt("missing type for model input '%s'", i.name.c_str()));
} else {
- throw MyError{fmt("invalid type for model input '%s': '%s'",
- i.name.c_str(), spec.c_str())};
+ throw MyError(fmt("invalid type for model input '%s': '%s'",i.name.c_str(), spec.c_str()));
}
}
bind_input(planner, i, input_type);