AI开发平台ModelArts-本地构建的模型如何转化为可应用Ascend芯片上的模型？:格式介绍

时间：2023-11-01 16:16:36

AI开发平台ModelArts

格式介绍

“caffemode”l格式

||---xxxx.caffemodel         模型参数文件，输入目录下有且只能有一个，必填。|---xxxx.prototxt           模型网络文件，输入目录下有且只能有一个，必填。|---insert_op_conf.cfg      插入算子配置文件，输入目录下有且只有一个，可选。|---plugin                  自定义算子目录，输入目录下有且只能有一个plugin文件夹，可选。仅支持基于TE（Tensor Engine）开发的自定义算子。

“frozen_graph”格式

||---xxxx.pb                 模型网络文件，输入目录下有且只能有一个，必填。支持以frozen_graph或saved_model格式保存的模型。|---insert_op_conf.cfg      插入算子配置文件，输入目录下有且只有一个，可选。|---plugin                  自定义算子目录，输入目录下有且只能有一个plugin文件夹，可选。仅支持基于TE（Tensor Engine）开发的自定义算子。

“saved_model”格式

||---saved_model.pb          模型网络文件，输入目录下有且只能有一个，必填。支持以frozen_graph或saved_model格式保存的模型。|---variables               固定子目录名称，包含模型的权重偏差等信息，必选    |---variables.index     必选    |---variables.data-00000-of-00001 必选|---insert_op_conf.cfg      插入算子配置文件，输入目录下有且只有一个，可选。|---plugin                  自定义算子目录，输入目录下有且只能有一个plugin文件夹，可选。仅支持基于TE（Tensor Engine）开发的自定义算子。

上一篇：AI开发平台ModelArts-在AI Gallery订阅的数据集可以在SDK中使用吗？

下一篇：AI开发平台ModelArts-在线服务和边缘服务有什么区别？