文档
注册

函数:set_model_dir

C函数原型

aclError aclopSetModelDir(const char *modelDir)

Python函数

ret = acl.op.set_model_dir(model_dir)

函数功能

设置加载模型文件的目录,该模型文件是由单算子编译成的*.om文件,同步接口。

输入说明

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 训练系列产品,在Host上运行应用时,一个进程内正在执行的算子的最大个数上限是40000000。

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

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

注意事项

参考资源

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

搜索结果
找到“0”个结果

当前产品无相关内容

未找到相关内容,请尝试其他搜索词