华为计算微信公众号
昇腾AI开发者公众号
华为计算微博
华为计算今日头条
将pytorch的模型转换为om模型后,我在pc端和板端进行了模型加载和推理。
我准备了1x2x1024x1024的numpy数据作为输入。
在pc端推理结果如下:
在板端推理,结果如下:
二者推理出的数据差别非常大。
而当我用2x512x512的numpy数据作为输入,pc和板端推理结果相同。
请问om模型的输入大小是否存在限制?
我要发帖子
将pytorch的模型转换为om模型后,我在pc端和板端进行了模型加载和推理。
我准备了1x2x1024x1024的numpy数据作为输入。
在pc端推理结果如下:
在板端推理,结果如下:
二者推理出的数据差别非常大。
而当我用2x512x512的numpy数据作为输入,pc和板端推理结果相同。
请问om模型的输入大小是否存在限制?