更新时间:2021-03-18 GMT+08:00
AIModelDescription
调用Init接口时作为输入,用于描述模型,详细实现参考ai_types.proto。
messAIPPDynaage AIModelDescription { string name = 1; // 模型名称,支持大小写字母、数字、下划线和点(.) int32 type = 2; // 模型类型,当前仅支持 DAVINCI_OFFLINE_MODEL类型,值为0 // 模型管家已新增模型解析能力,因此该字段不设置亦无碍,为了保持向前兼容,特此保留--2018/11/24 string version = 3; // 模型版本 int32 size = 4; // 模型大小 string path = 5; // 模型路径,支持大小写字母、数字、下划线 repeated string sub_path = 6; // 辅助模型路径,用于模型文件为多个的情况,比如caffe在线模型 // 为了向前兼容,我们没有对path字段做修改(修改为repeated),而是新增了sub_path字段 string key = 7; // 模型秘钥 repeated string sub_key = 8; // 辅助模型秘钥,用于模型秘钥为多个的情况,比如caffe在线模型 // 为了向前兼容,我们没有对key字段做修改(修改为repeated),而是新增了sub_key字段 enum Frequency { UNSET =0; LOW =1; MEDIUM =2; HIGH =3; } Frequency frequency = 9; enum DeviceType { NPU = 0; IPU = 1; MLU = 2; CPU = 3; NONE = 255; } DeviceType device_type = 10; enum Framework { OFFLINE =0; CAFFE =1; TENSORFLOW =2; } Framework framework = 11; bytes data = 100; // 模型数据 repeated AITensorDescription inputs = 12; // 输入Tensor描述 repeated AITensorDescription outputs = 13; // 输出Tensor描述 };
父主题: 数据类型