不同输入尺寸的engine模型转换报错

你好，我需要 engine 模型接受指定尺寸的输入，pt转wts正常。然后修改配置文件中的kInputH和kInputW为自己的尺寸，wts转engine模型报错了，请问还需要修改其他的地方吗？谢谢