使用mindspore将pkl文件转为onnx时报错

问题描述:
【操作步骤&问题现象】
使用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将pkl文件转为onnx时报错
文章图片



【使用mindspore将pkl文件转为onnx时报错】具体参见 mindspore/mindspore/ccsrc/transform/express_ir/onnx_exporter.cc

    推荐阅读