问题描述:
【操作步骤&问题现象】
使用mindspore将pkl文件转为onnx时报错,提示“RuntimeError: mindspore/ccsrc/transform/express_ir/onnx_exporter.cc:1657 ExportPrimitive] Can not find key NotEqual in convert map. Exporting NotEqual operator is not yet supported,”
【具体代码】
from mindspore import Tensor, Model,export,load_checkpoint
model_name = 'wukong_vit_b_32_clip'
config_path = './src/config/{}.yaml'.format(model_name)
text_encoder = load_text_model(config['model']['text'],'/wukong_vit_b_32_clip.pkl')
text_encoder = text_encoder.to_float(mstype.float16)
export(text_encoder, Tensor(tokenize('火焰')), file_name='text_encoder.onnx', file_format='ONNX')
解答:
有不支持的算子,目前支持的转换算子为
文章图片
文章图片
【使用mindspore将pkl文件转为onnx时报错】具体参见 mindspore/mindspore/ccsrc/transform/express_ir/onnx_exporter.cc
推荐阅读
- 大数据|阿里巴巴达摩院发布2021十大科技趋势!
- 大数据|数据读取与保存Sequence文件_大数据培训
- spark|Spark SQL的自定义函数UDF
- 开源流式湖仓服务 Arctic 详解(并非另一套 Table Format)
- 数据仓库|数据仓库建设方案详细(系统架构设计)
- 数据仓库|Hive、SparkSQL是如何决定写文件的数量的()
- 大数据|大数据技术面试-项目架构与涉及技术
- 数据仓库|数据仓库完整版
- 数据建模|数据质量维度详解