昇腾社区首页
中文
注册

函数:get_desc_from_file

产品支持情况

产品

是否支持

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

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

Atlas 训练系列产品

Atlas 推理系列产品

Atlas 200I/500 A2 推理产品

Atlas 200/300/500 推理产品

功能说明

根据模型文件获取该模型的模型描述信息。

函数原型

  • C函数原型
    1
    aclError aclmdlGetDescFromFile(aclmdlDesc *modelDesc, const char *modelPath)
    
  • python函数
    1
    ret = acl.mdl.get_desc_from_file(model_desc, model_path)
    

参数说明

参数名

说明

model_desc

int,aclmdlDesc类型的指针。需提前调用create_desc接口创建aclmdlDesc类型的数据,

调用成功后,会刷新指针地址里面的内容。

model_path

str,离线模型文件的存储路径,路径中包含文件名。运行程序(APP)的用户需要对该存储路径有访问权限。

此处的离线模型文件是适配昇腾AI处理器的离线模型,即*.om文件。

说明:
  • 关于如何获取om文件,请参见模型构建
  • om模型文件大小有限制的场景下,使用ATC工具生成om文件时,如将“--external_weight”参数设置为11表示将原始网络中的Const/Constant节点的权重保存在单独的文件中,且该文件保存在与om文件同级的“weight”目录下),那么在使用本接口加载om文件时,需将“weight”目录与om文件放在同级目录下,会进入“weight”目录查找权重文件,未正确放置“weight”目录可能会导致单独的权重文件加载不成功。

返回值说明

返回值

说明

ret

int,返回0表示成功,返回非0表示失败。

约束说明

通过本接口获取到的模型描述信息,无法应用于get_op_attrget_cur_output_dims接口。