华为云AI开发平台ModelArts模型转换输入目录规范_云淘科技
模型转换时,模型输入目录有一定规范,请按照如下要求说明,将您的模型包按照要求上传至OBS。
Ascend芯片
用于Ascend芯片的模型,其转换要求如下所示:
针对基于Caffe框架的模型,执行模型转换时,其输入目录需符合如下规范。
| |---xxxx.caffemodel 模型参数文件,输入目录下有且只能有一个,必填。 |---xxxx.prototxt 模型网络文件,输入目录下有且只能有一个,必填。 |---insert_op_conf.cfg 插入算子配置文件,输入目录下有且只有一个,可选。 |---plugin 自定义算子目录,输入目录下有且只能有一个plugin文件夹,可选。仅支持基于TE(Tensor Engine)开发的自定义算子。
针对基于TensorFlow框架的模型(“frozen_graph”或“saved_model”格式),执行模型转换时,其输入目录需符合如下规范。
“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)开发的自定义算子。
父主题: 模型转换
同意关联代理商云淘科技,购买华为云产品更优惠(QQ 78315851)
内容没看懂? 不太想学习?想快速解决? 有偿解决: 联系专家