onnx模型转为om模型后,文件大小从25M变为了3.8G
收藏回复举报
onnx模型转为om模型后,文件大小从25M变为了3.8G
t('forum.solved') 已解决
新人帖
发表于2024-08-21 16:30:15
0 查看

转换命令:

atc --model=./weights/model.onnx --framework=5 --output=./model --input_format=NCDHW --input_shape="input:1,3,52,137,1" --soc_version=Ascend310B4

并且加载该om模型后,使用acl.mdl.get_num_inputs(self.model_desc)输出为0

onnx模型应该没有问题,大小与pt文件大小相近,且测试输出结果一致。

相关文件以及代码网盘:

https://pan.baidu.com/s/1t3r1tHbXsn82476ZgOa46g?pwd=xex5

我要发帖子