如何部署这个模型？

raoyongming / GFNet

[NeurIPS 2021] [T-PAMI] Global Filter Networks for Image Classification

https://gfnet.ivg-research.xyz/

MIT License

450 stars 42 forks source link

Closed JiangWeiHn closed 2 years ago

JiangWeiHn commented 2 years ago

您好，我想部署hornet在Nvidia Xavier上，用到这个GFNet,但在转onnx时，遇到torch.tff不支持的情况，有什么好的解决办法么？

raoyongming commented 2 years ago

你好，感谢对我们工作的关注。我们没有onnx部署的经验，我找到这里提到了onnx支持fft相关操作。如果不支持rfft/irfft可以直接用fft/ifft替代，但可能会增加少量的额外计算。另外对于HorNet，也可以使用7x7版本的模型。

JiangWeiHn commented 2 years ago

好的谢谢