5. 使用 ONNXRuntime 进行预测

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
}