如何导出NAFNetLocal模型文件为onnx格式？

[lmw0320]

如图所示，我尝试将这个模型文件导出onnx文件，实际设置为动态尺寸输入时，会报错：

我检查了NAFNet网络中的check_image_size函数，并增加了打印信息，用于查验:

不知道这个模型，到底要如何导出呢？？
----我个人测试过，如果模型事先设定一张图片的尺寸数据，导出onnx文件后。可以正常调用onnx，对该图片进行 推理，结果也正常。而使用另外一张不同尺寸的图片进行onnx模型推理时，就会报错，提示输入的数据形状不对。。---因此，我才希望导出一个可以支持动态尺寸输入的onnx文件，以便进行正常应用。