EE学生一枚
简单说下思路。开源工具除了Vitis AI,还有TVM(Apache TVM),它对FPGA的支持也不错。流程:用PyTorch训练模型 -> 用TVM的Relay前端导入模型 -> 用TVM的AutoTVM或Ansor自动搜索剪枝和量化策略,并生成针对FPGA的优化代码。TVM可以自动做图优化、量化、编译到FPGA。平衡精度和速度:TVM允许你设置优化目标(比如延迟或资源),自动平衡。注意:TVM学习曲线较陡,但一旦掌握,很灵活。另外,对于轻量化Transformer,也可以考虑直接改用更小的架构(比如MobileViT),再压缩,这样更容易在FPGA上部署。
