AI开发平台MODELARTS-模型转换操作:约束限制

时间:2024-04-17 15:54:08

约束限制

  • 模型转换当前只支持Ascend芯片类型。
  • 模型转换当前仅支持原始框架类型为Caffe和Tensorflow的模型转换。当原始框架类型为Caffe时,输入数据类型为FLOAT;当原始框架类型为Tensorflow时,输入数据类型为INT32、BOOL、UINT8、FLOAT。
  • ModelArts提供了转换模板供用户选择,只能选择对应模板进行转换,支持的模板描述,请参见转换模板
  • 针对用于Ascend芯片的模型转换,其转换限制说明可参见“昇腾开发者社区”约束及参数说明
  • 压缩/转换任务指定的OBS路径,需确保OBS目录与ModelArts在同一区域。
  • 转换后的模型,再导入ModelArts时,需从模板中选择元模型
  • 支持模型文件类型为onnx的模型转换,转换时会先将其转换为Tensorflow框架的FrozenGraphDef格式,然后再转换为om格式。转换工具要求onnx版本为1.6.0,opset为9+。
  • 当原始框架类型为Caffe时,模型文件(“.prototxt”)和权重文件(“.caffemodel”)的op name、op type必须保持名称一致(包括大小写)。
  • 当原始框架类型为Caffe时,除了top与bottom相同的layer以外(例如BatchNorm,Scale,ReLU等),其他layer的top名称需要与其name名称保持一致。
  • 当原始框架类型为TensorFlow时,支持FrozenGraphDef格式和SavedModel格式。如果是SavedModel格式,转换时会先将其转换为FrozenGraphDef格式,然后再转换为om格式。
  • 不支持动态shape的输入,例如:NHWC输入为[?,?,?,3]多个维度可任意指定数值。模型转换时需指定固定数值。
  • 输入数据最大支持四维,转维算子(reshape、expanddim等)不能输出五维。
  • 模型中的所有层算子除const算子外,输入和输出需要满足“dim!=0”
  • 模型转换不支持含有训练算子的模型。
  • 量化(uint8)后的模型不支持模型转换。
  • 模型中的算子只支持2D卷积,暂不支持3D卷积。暂不支持多批量转换batch_normalization_1算子和FusedBatchNorm算子。
  • 只支持Caffe算子清单Tensorflow算子清单中的算子,并需满足算子限制条件。
support.huaweicloud.com/develop-modelarts/develop-modelarts-0060.html