om模型的输入大小是否存在限制?
收藏回复举报
om模型的输入大小是否存在限制?
t('forum.solved') 已解决
新人帖
发表于2024-10-19 15:51:15
0 查看

将pytorch的模型转换为om模型后,我在pc端和板端进行了模型加载和推理。

我准备了1x2x1024x1024的numpy数据作为输入。

在pc端推理结果如下:

cke_7384.png

在板端推理,结果如下:

cke_10777.png

二者推理出的数据差别非常大。

而当我用2x512x512的numpy数据作为输入,pc和板端推理结果相同。

请问om模型的输入大小是否存在限制?

我要发帖子