使用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/mindspore/ccsrc/transform/express_ir/onnx_exporter.cc