更新时间:2024-11-21 GMT+08:00
转换输入目录
解释说明
模型转换时,待转换模型存放的OBS路径。基于不同框架的模型,其模型输入目录的要求不同。
- 输入框架为Caffe
基于Caffe框架的模型,执行模型转换时,其输入目录需符合如下规范。
| |---xxxx.caffemodel 模型参数文件,输入目录下有且只能有一个,必填。 |---xxxx.prototxt 模型网络文件,输入目录下有且只能有一个,必填。 |---insert_op_conf.cfg 插入算子配置文件,输入目录下有且只有一个,可选。 |---plugin 自定义算子目录,输入目录下有且只能有一个plugin文件夹,可选。仅支持基于TE(Tensor Engine)开发的自定义算子。
- 输入框架为TensorFlow
基于TensorFlow框架的模型,执行模型转换时,其输入目录需符合如下规范。
| |---saved_model.pb 模型网络文件,输入目录下有且只能有一个,必填。支持以frozen_graph或saved_model格式保存的模型。 |---variables 固定子目录名称,包含模型的权重偏差等信息。“frozen_graph”格式不需要,“saved_model”格式必选。 |---variables.index 必选 |---variables.data-00000-of-00001 必选 |---insert_op_conf.cfg 插入算子配置文件,输入目录下有且只有一个,可选。 |---plugin 自定义算子目录,输入目录下有且只能有一个plugin文件夹,可选。仅支持基于TE(Tensor Engine)开发的自定义算子。