API定义如下:
// 启用 ONNXRuntime 进行预测 // 参数:None // 返回:None func (config *Config) EnableONNXRuntime() // 禁用 ONNXRuntime 进行预测 // 参数:None // 返回:None func (config *Config) DisableONNXRuntime(); // 判断是否启用 ONNXRuntime // 参数:None // 返回:bool - 是否启用 ONNXRuntime func (config *Config) ONNXRuntimeEnabled() bool; // 启用 ONNXRuntime 预测时开启优化 // 参数:None // 返回:None func (config *Config) EnableORTOptimization();
ONNXRuntime设置代码示例:
package main
// 引入 Paddle Golang Package
import pd "github.com/paddlepaddle/paddle/paddle/fluid/inference/goapi"
import fmt
func main() {
// 创建 Config 对象
config := pd.NewConfig()
// 启用 ONNXRuntime 进行预测
config.EnableONNXRuntime()
// 通过 API 获取 ONNXRuntime 信息
fmt.Println("Use ONNXRuntime is: ", config.ONNXRuntimeEnabled()) // True
// 开启ONNXRuntime优化
config.EnableORTOptimization();
// 禁用 ONNXRuntime 进行预测
config.DisableONNXRuntime()
// 通过 API 获取 ONNXRuntime 信息
fmt.Println("Use ONNXRuntime is: ", config.ONNXRuntimeEnabled()) // False
}