HLS 加速卷积神经网络前向计算(毕设小结)

       本文主要是本人本科毕业设计的主要工作。        主要工作有两部分,一是实现二值卷积神经网络模型,二是将模型的前向计算部署到FPGA上。也就是说FPGA负责的是神经网络的前向计算过程,也就是预测的过程,训练还是通过CPU。        二值卷积神经网络是参考复现XNOR-Net,训练集就采取了最简单的mnist,实现方法是采用的pytorch,HLS中的前向计算是通过C++编写
相关文章
相关标签/搜索