AI开发平台MODELARTS-SD1.5基于DevServer适配MindSpore-Lite NPU推理指导(6.3.T041):附录:模型转换样例参考

时间:2024-07-06 17:52:03

附录:模型转换样例参考

cd ***/onnx/unet source /usr/local/Ascend/ascend-toolkit/set_env.sh && converter_lite --modelFile=./model.onnx --outputFile=./unet_test --fmk=ONNX --saveType=MINDIR --optimize=ascend_oriented --inputShape="sample:2,4,64,64;timestep:1;encoder_hidden_states:2,77,768"

onnx模型本身是动态的,这里以图片尺寸为512*512时,unet的输入shape为例。

更多模型转换的详细信息可以参考MindSpore Lite的官方文档:https://www.mindspore.cn/lite/docs/zh-CN/r2.3.0rc1/use/cloud_infer/converter_tool.html

support.huaweicloud.com/bestpractice-modelarts/modelarts_10_1503.html