MSConfig
import com.mindspore.lite.config.MSConfig;
MSConfig类用于保存执行中的配置变量。
公有成员函数
function |
---|
boolean init(int deviceType, int threadNum, int cpuBindMode, boolean enable_float16) |
boolean init(int deviceType, int threadNum, int cpuBindMode) |
init
public boolean init(int deviceType, int threadNum, int cpuBindMode, boolean enable_float16)
初始化MSConfig。
参数
deviceType
: 设备类型,deviceType
在com.mindspore.lite.config.DeviceType中定义。threadNum
: 线程数。cpuBindMode
: CPU绑定模式,cpuBindMode
在com.mindspore.lite.config.CpuBindMode中定义。enable_float16
:是否优先使用float16算子。
返回值
初始化是否成功。
public boolean init(int deviceType, int threadNum, int cpuBindMode)
初始化MSConfig,enable_float16
默认为false。
参数
deviceType
: 设备类型,deviceType
在com.mindspore.lite.config.DeviceType中定义。threadNum
: 线程数。cpuBindMode
: CPU绑定模式,cpuBindMode
在com.mindspore.lite.config.CpuBindMode中定义。
返回值
初始化是否成功。
public boolean init(int deviceType, int threadNum)
初始化MSConfig,cpuBindMode
默认为CpuBindMode.MID_CPU
,enable_float16
默认为false。
参数
deviceType
: 设备类型,deviceType
在com.mindspore.lite.config.DeviceType中定义。threadNum
: 线程数。
返回值
初始化是否成功。
public boolean init(int deviceType)
初始化MSConfig,cpuBindMode
默认为CpuBindMode.MID_CPU
,threadNum
默认为2
,enable_float16
默认为false。
参数
deviceType
: 设备类型,deviceType
在com.mindspore.lite.config.DeviceType中定义。
返回值
初始化是否成功。
public boolean init()
初始化MSConfig,deviceType
默认为DeviceType.DT_CPU
,cpuBindMode
默认为CpuBindMode.MID_CPU
,threadNum
默认为2
,enable_float16
默认为false。
返回值
初始化是否成功。
free
public void free()
释放MSConfig运行过程中动态分配的内存。LiteSession init之后即可释放。