From f57c056ccdf207ebe07fc24ed99314c3e4863093 Mon Sep 17 00:00:00 2001 From: Yuya Hanai Date: Fri, 21 Jan 2022 19:25:32 +0900 Subject: [PATCH] change memory info handler --- src/ofxOnnxRuntime.h | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/ofxOnnxRuntime.h b/src/ofxOnnxRuntime.h index e8001f5..9e7e634 100644 --- a/src/ofxOnnxRuntime.h +++ b/src/ofxOnnxRuntime.h @@ -37,7 +37,7 @@ namespace ofxOnnxRuntime std::vector input_node_dims; // 1 input only. std::size_t input_tensor_size = 1; std::vector input_values_handler; - Ort::MemoryInfo memory_info_handler = Ort::MemoryInfo::CreateCpu(OrtDeviceAllocator, OrtMemTypeCPU); + Ort::MemoryInfo memory_info_handler = Ort::MemoryInfo::CreateCpu(OrtArenaAllocator, OrtMemTypeDefault); std::vector output_node_names; std::vector> output_node_dims; // >=1 outputs std::vector output_values;