昇腾社区首页
中文
注册

函数:set_model_dir

产品支持情况

产品

是否支持

Atlas A3 训练系列产品 / Atlas A3 推理系列产品

Atlas A2 训练系列产品 / Atlas 800I A2 推理产品 /A200I A2 Box 异构组件

Atlas 训练系列产品

Atlas 推理系列产品

Atlas 200I/500 A2 推理产品

Atlas 200/300/500 推理产品

功能说明

设置加载模型文件的目录,目录下存放单算子模型文件(*.om文件)。

函数原型

  • C函数原型
    1
    aclError aclopSetModelDir(const char *modelDir)
    
  • python函数
    1
    ret = acl.op.set_model_dir(model_dir)
    

参数说明

参数名

说明

model_dir

str,表示指定模型文件所在目录。

此处可设置多级目录,但系统最多从多级目录的最后一级开始,读取三级目录下的模型文件。

例如,将modelDir设置为"dir0/dir1",dir1下的目录层级为“dir2/dir3/dir4”,这时系统只支持读取“dir1”“dir1/dir2”“dir1/dir2/dir3”目录下的模型文件。

返回值说明

返回值

说明

ret

int,错误码,返回0表示成功,返回其它值表示失败。

约束说明

在加载前,请先根据文件的大小评估内存空间是否足够,内存空间不足,会导致应用程序异常。

Atlas 200/300/500 推理产品 ,在Host上运行应用时,一个进程内正在执行的算子的最大个数上限是40000000,在Device上运行应用时,一个进程内正在执行的算子的最大个数上限是2000000。

Atlas 200I/500 A2 推理产品 ,在Host上运行应用时,一个进程内正在执行的算子的最大个数上限是40000000。

Atlas 训练系列产品 ,在Host上运行应用时,一个进程内正在执行的算子的最大个数上限是40000000。

Atlas A2 训练系列产品 / Atlas 800I A2 推理产品 /A200I A2 Box 异构组件,在Host上运行应用时,一个进程内正在执行的算子的最大个数上限是2000000。

Atlas 推理系列产品 ,在Host上运行应用时,一个进程内正在执行的算子的最大个数上限是40000000,在Device上运行应用时,一个进程内正在执行的算子的最大个数上限是2000000。

Atlas A3 训练系列产品 / Atlas A3 推理系列产品 ,在Host上运行应用时,一个进程内正在执行的算子的最大个数上限是2000000。

资源参考

接口调用流程、示例,参见单算子调用