函数: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,错误码。
|
约束说明 |
在加载前,请先根据文件的大小评估内存空间是否足够,内存空间不足,会导致应用程序异常。 Atlas 200/300/500 推理产品,在Host上运行应用时,一个进程内正在执行的算子的最大个数上限是40000000,在Device上运行应用时,一个进程内正在执行的算子的最大个数上限是2000000。 Atlas 200I/500 A2推理产品,在Host上运行应用时,一个进程内正在执行的算子的最大个数上限是40000000。 Atlas 训练系列产品,在Host上运行应用时,一个进程内正在执行的算子的最大个数上限是40000000。 Atlas A2训练系列产品/Atlas 800I A2推理产品,在Host上运行应用时,一个进程内正在执行的算子的最大个数上限是2000000。 Atlas 推理系列产品(Ascend 310P处理器),在Host上运行应用时,一个进程内正在执行的算子的最大个数上限是40000000,在Device上运行应用时,一个进程内正在执行的算子的最大个数上限是2000000。 |
参考资源 |