HLS 加速卷積神經網絡前向計算(畢設小結)

       本文主要是本人本科畢業設計的主要工作。        主要工作有兩部分,一是實現二值卷積神經網絡模型,二是將模型的前向計算部署到FPGA上。也就是說FPGA負責的是神經網絡的前向計算過程,也就是預測的過程,訓練還是通過CPU。        二值卷積神經網絡是參考復現XNOR-Net,訓練集就採取了最簡單的mnist,實現方法是採用的pytorch,HLS中的前向計算是通過C++編寫
相關文章
相關標籤/搜索