华为云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)

内容没看懂? 不太想学习?想快速解决? 有偿解决: 联系专家