From f4f2aef534cd95497476685cacc8f0970e5a3183 Mon Sep 17 00:00:00 2001 From: tuna2134 Date: Mon, 9 Sep 2024 09:55:50 +0000 Subject: [PATCH] fix: level --- sbv2_core/src/bert.rs | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/sbv2_core/src/bert.rs b/sbv2_core/src/bert.rs index a5c30b2..3377bdc 100644 --- a/sbv2_core/src/bert.rs +++ b/sbv2_core/src/bert.rs @@ -4,7 +4,7 @@ use ort::{GraphOptimizationLevel, Session}; pub fn load_model() -> Result { let session = Session::builder()? - .with_optimization_level(GraphOptimizationLevel::Level1)? + .with_optimization_level(GraphOptimizationLevel::Level3)? .with_intra_threads(1)? .commit_from_file("models/debert.onnx")?; Ok(session)