使用pytorch训练图像分割模型，模型转换图像大小问题

请问在X3派上部署自己训练的图像分割模型，模型转换使用图像大小必须是1024*2048大小吗？我修改模型量化的图像大小需要在哪里进行修改呢？我自己训练的模型图像大小是512*512，使用的框架是基于训练的。